Method test_load_dataset

tests/test_hf_datasets.py:18–21

(self)

        self.assertEqual('foo', mapped_df[0]['label'])
        
    def test_load_dataset(self):
        dataset = datasets.load_dataset("csv", data_files="/input/tests/data/train.csv")
        full_data = pd.DataFrame(dataset['train'])
        self.assertFalse(full_data.empty)