Add DataFrame creation and filtering test

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():
+    for dataset in DATASETS:
-        df = create_dataframe(file)
+        request_dataset(dataset)
+        df = create_dataframe(dataset)
+        remove(FILES[dataset])
         assert isinstance(df, DataFrame)
+        assert list(df) == COLUMNS[dataset]