Add dataframe creation test
This commit is contained in:
0
tests/__init__.py
Normal file
0
tests/__init__.py
Normal file
9
tests/preprocessing_test.py
Normal file
9
tests/preprocessing_test.py
Normal file
@@ -0,0 +1,9 @@
|
||||
from app.constants import files
|
||||
from app.preprocessing import create_dataframe
|
||||
from pandas import DataFrame
|
||||
|
||||
|
||||
def test_dataframe_creation():
|
||||
for file in files.keys():
|
||||
df = create_dataframe(file)
|
||||
assert isinstance(df, DataFrame)
|
||||
Reference in New Issue
Block a user