Method test_EvalSidDataset_initialization

data_test.py:195–207

Test SidItemFeatDataset initialization

(self)

        self.assertTrue(hasattr(dataset, 'inputs'))
        
    def test_EvalSidDataset_initialization(self):
        """Test SidItemFeatDataset initialization"""
        dataset = EvalSidDataset(
            train_file=self.csv_file,
            tokenizer=self.tokenizer,
            max_len=128,
            sample=1,
            seed=0,
            category="games"
        )

        self.assertEqual(len(dataset), 1)
        self.assertTrue(hasattr(dataset, 'inputs'))

    def test_RLTitle2SidDataset_initialization(self):
        """Test RLTitle2SidDataset initialization"""