coolneng/graphPaname
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add DataFrame creation and filtering test

Browse Source
This commit is contained in:
coolneng 2020-06-10 21:50:08 +02:00
parent 53a34701da
commit 74ce88e8b3
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/preprocessing_test.py
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)