Test D3Dataset initialization
(self)
| 117 | self.assertTrue(hasattr(dataset, 'inputs')) |
| 118 | |
| 119 | def test_D3Dataset_initialization(self): |
| 120 | """Test D3Dataset initialization""" |
| 121 | dataset = D3Dataset( |
| 122 | train_file=self.csv_file, |
| 123 | max_len=128, |
| 124 | sample=1, |
| 125 | seed=0, |
| 126 | category="games" |
| 127 | ) |
| 128 | self.assertEqual(len(dataset), 1) |
| 129 | self.assertTrue(hasattr(dataset, 'inputs')) |
| 130 | |
| 131 | def test_EvalD3Dataset_initialization(self): |
| 132 | """Test EvalD3Dataset initialization""" |