Compare commits
5 Commits
87fa53f6f8
...
1c1e5ba4dc
Author | SHA1 | Date | |
---|---|---|---|
1c1e5ba4dc | |||
5a1882dfa2 | |||
6d5ba4de25 | |||
4173a43b90 | |||
53a34701da |
@ -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]
|
||||||
|
Loading…
Reference in New Issue
Block a user