diff --git a/tests/preprocessing_test.py b/tests/preprocessing_test.py index b032f5b..a01cd61 100644 --- a/tests/preprocessing_test.py +++ b/tests/preprocessing_test.py @@ -1,9 +1,21 @@ -from app.constants import files -from app.preprocessing import create_dataframe from pandas import DataFrame +from requests import get +from app.constants import URL, COLUMNS, FILES, DATASETS +from app.request_datasets import request_dataset +from app.preprocessing import create_dataframe +from os import remove + + +def test_dataset_request(): + for dataset in DATASETS: + response = get(URL.format(dataset)) + assert response.status_code == 200 def test_dataframe_creation(): - for file in files.keys(): - df = create_dataframe(file) + for dataset in DATASETS: + request_dataset(dataset) + df = create_dataframe(dataset) + remove(FILES[dataset]) assert isinstance(df, DataFrame) + assert list(df) == COLUMNS[dataset]