|
|
@ -109,29 +109,6 @@ class TestMultiSlotStringDataGenerator(unittest.TestCase):
|
|
|
|
my_ms_dg.run_from_memory()
|
|
|
|
my_ms_dg.run_from_memory()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestMultiSlotStringDataGenerator_2(unittest.TestCase):
|
|
|
|
|
|
|
|
def test_MyMultiSlotStringDataGenerator_stdin(self):
|
|
|
|
|
|
|
|
plats = platform.platform()
|
|
|
|
|
|
|
|
if 'Linux' not in plats:
|
|
|
|
|
|
|
|
print("skip pipecommand UT on MacOS/Win")
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
with open("test_queue_dataset_run_a.txt", "w") as f:
|
|
|
|
|
|
|
|
data = "2 1 2\n"
|
|
|
|
|
|
|
|
data += "2 6 2\n"
|
|
|
|
|
|
|
|
data += "2 5 2\n"
|
|
|
|
|
|
|
|
data += "2 7 2\n"
|
|
|
|
|
|
|
|
f.write(data)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
tmp = os.popen(
|
|
|
|
|
|
|
|
"cat test_queue_dataset_run_a.txt | python my_data_generator.py"
|
|
|
|
|
|
|
|
).readlines()
|
|
|
|
|
|
|
|
expected_res = [
|
|
|
|
|
|
|
|
'1 2 1 1 1 2\n', '1 2 1 6 1 2\n', '1 2 1 5 1 2\n', '1 2 1 7 1 2\n'
|
|
|
|
|
|
|
|
]
|
|
|
|
|
|
|
|
self.assertEqual(tmp, expected_res)
|
|
|
|
|
|
|
|
os.remove("./test_queue_dataset_run_a.txt")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestMultiSlotDataGenerator_error(unittest.TestCase):
|
|
|
|
class TestMultiSlotDataGenerator_error(unittest.TestCase):
|
|
|
|
def test_MultiSlotDataGenerator_error(self):
|
|
|
|
def test_MultiSlotDataGenerator_error(self):
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|
with self.assertRaises(ValueError):
|
|
|
|