[#41] update test case for read raw datasets

aisingapore · Jan 13, 2022 · fa79413 · fa79413
1 parent 714c0bc
commit fa79413
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/sentic_gcn/test_sentic_gcn_utils.py b/tests/sentic_gcn/test_sentic_gcn_utils.py
@@ -101,7 +101,7 @@ def test_read_raw_dataset(self):
         with mock.patch("sgnlp.models.sentic_gcn.tokenization.SenticGCNTokenizer") as MockClass:
             fake_tokenizer = MockClass()
         dataset_gen = SenticGCNDatasetGenerator(self.cfg, fake_tokenizer)
-        data = dataset_gen._read_raw_dataset("train")
+        data = dataset_gen._read_raw_dataset(self.cfg.dataset_train)
         self.assertEqual(len(data), 15)
 
     def test_generate_senticgcn_dataset(self):