tests/test_datasets.py

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()