| import unittestfrom data.datasets import load_dataset, preprocess_dataclass TestDatasets(unittest.TestCase): def test_load_dataset(self): df = load_dataset('data/test.csv') self.assertIsNotNone(df) def test_preprocess_data(self): df = load_dataset('data/test.csv') preprocessed_df = preprocess_data(df) self.assertFalse(preprocessed_df.isnull().values.any())if __name__ == '__main__': unittest.main() | |