Compare commits

..

1 Commits

Author SHA1 Message Date
4173a43b90
Add DataFrame creation and filtering test 2020-06-10 21:56:55 +02:00

View File

@ -7,12 +7,18 @@ from os import remove
def test_dataset_request(): def test_dataset_request():
"""
Checks that the datasets URLs are reachable
"""
for dataset in DATASETS: for dataset in DATASETS:
response = get(URL.format(dataset)) response = get(URL.format(dataset))
assert response.status_code == 200 assert response.status_code == 200
def test_dataframe_creation(): def test_dataframe_creation():
"""
Verifes that the DataFrames are created and filtered properly
"""
for dataset in DATASETS: for dataset in DATASETS:
request_dataset(dataset) request_dataset(dataset)
df = create_dataframe(dataset) df = create_dataframe(dataset)