Add dataframe column assertion

tests/preprocessing_test.py

@@ -1,9 +1,11 @@
-from constants import FILES, DATASETS, URL
-from app.preprocessing import create_dataframe
-from app.request_datasets import request_dataset
+from os import remove
+
 from pandas import DataFrame
 from requests import get
-from os import remove
+
+from app.preprocessing import create_dataframe
+from app.data_request import request_dataset
+from constants import COLUMNS, DATASETS, FILES, URL
 
 
 def test_dataset_request():
@@ -24,3 +26,4 @@ def test_dataframe_creation():
         df = create_dataframe(dataset)
         remove(FILES[dataset])
         assert isinstance(df, DataFrame)
+        assert all(df.columns == COLUMNS[dataset])