tests/test_datasets.py

import unittest
from data.datasets import load_dataset, preprocess_data

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