Update tests/test_raw_text_inference.py
#2
by minhgou - opened
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": "
|
| 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": "
|
| 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": "
|
| 188 |
with self.assertRaises(ValueError):
|
| 189 |
bsc_runner.generate_from_chinese_text(" ")
|
| 190 |
|
| 191 |
celer_runner = object.__new__(EyettentionRawTextInference)
|
| 192 |
-
celer_runner.cf = {"dataset": "
|
| 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 |
|