Update tests/test_raw_text_inference.py

#2
by minhgou - opened
Files changed (1) hide show
  1. tests/test_raw_text_inference.py +4 -4
tests/test_raw_text_inference.py CHANGED
@@ -171,12 +171,12 @@ class RawTextInferenceSmokeTests(unittest.TestCase):
171
  from Eyettention.raw_text_inference import EyettentionRawTextInference
172
 
173
  bsc_runner = object.__new__(EyettentionRawTextInference)
174
- bsc_runner.cf = {"dataset": "celer"}
175
  with self.assertRaises(ValueError):
176
  bsc_runner.generate_from_chinese_text(None)
177
 
178
  celer_runner = object.__new__(EyettentionRawTextInference)
179
- celer_runner.cf = {"dataset": "BSC"}
180
  with self.assertRaises(ValueError):
181
  celer_runner.generate_from_english_text(None)
182
 
@@ -184,12 +184,12 @@ class RawTextInferenceSmokeTests(unittest.TestCase):
184
  from Eyettention.raw_text_inference import EyettentionRawTextInference
185
 
186
  bsc_runner = object.__new__(EyettentionRawTextInference)
187
- bsc_runner.cf = {"dataset": "celer"}
188
  with self.assertRaises(ValueError):
189
  bsc_runner.generate_from_chinese_text(" ")
190
 
191
  celer_runner = object.__new__(EyettentionRawTextInference)
192
- celer_runner.cf = {"dataset": "BSC"}
193
  with self.assertRaises(ValueError):
194
  celer_runner.generate_from_english_text(" ")
195
 
 
171
  from Eyettention.raw_text_inference import EyettentionRawTextInference
172
 
173
  bsc_runner = object.__new__(EyettentionRawTextInference)
174
+ bsc_runner.cf = {"dataset": "BSC"}
175
  with self.assertRaises(ValueError):
176
  bsc_runner.generate_from_chinese_text(None)
177
 
178
  celer_runner = object.__new__(EyettentionRawTextInference)
179
+ celer_runner.cf = {"dataset": "celer"}
180
  with self.assertRaises(ValueError):
181
  celer_runner.generate_from_english_text(None)
182
 
 
184
  from Eyettention.raw_text_inference import EyettentionRawTextInference
185
 
186
  bsc_runner = object.__new__(EyettentionRawTextInference)
187
+ bsc_runner.cf = {"dataset": "BSC"}
188
  with self.assertRaises(ValueError):
189
  bsc_runner.generate_from_chinese_text(" ")
190
 
191
  celer_runner = object.__new__(EyettentionRawTextInference)
192
+ celer_runner.cf = {"dataset": "celer"}
193
  with self.assertRaises(ValueError):
194
  celer_runner.generate_from_english_text(" ")
195