Compare commits

..

2 Commits

Author SHA1 Message Date
74ce88e8b3
Add DataFrame creation and filtering test 2020-06-10 22:10:18 +02:00
53a34701da
Add DataFrame creation and filtering test 2020-06-10 21:50:08 +02:00

View File

@ -1,4 +1,7 @@
from app.constants import files 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 app.preprocessing import create_dataframe
from os import remove from os import remove
@ -21,3 +24,4 @@ def test_dataframe_creation():
df = create_dataframe(dataset) df = create_dataframe(dataset)
remove(FILES[dataset]) remove(FILES[dataset])
assert isinstance(df, DataFrame) assert isinstance(df, DataFrame)
assert list(df) == COLUMNS[dataset]