Test SidDataset initialization
(self)
| 142 | self.assertTrue(hasattr(dataset, 'inputs')) |
| 143 | |
| 144 | def test_SidDataset_initialization(self): |
| 145 | """Test SidDataset initialization""" |
| 146 | dataset = SidDataset( |
| 147 | train_file=self.csv_file, |
| 148 | max_len=128, |
| 149 | sample=1, |
| 150 | seed=0, |
| 151 | category="games" |
| 152 | ) |
| 153 | self.assertEqual(len(dataset), 1) |
| 154 | self.assertTrue(hasattr(dataset, 'inputs')) |
| 155 | |
| 156 | def test_SidSFTDataset_initialization(self): |
| 157 | """Test SidSFTDataset initialization""" |