graphPaname/tests/preprocessing_test.py

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)