diff --git a/python/paddle/fluid/tests/unittests/my_data_generator.py b/python/paddle/fluid/tests/unittests/my_data_generator.py index 6d5061edf6b6d255ce6d449b379032c9e19096d7..ac906b3256163f421310f10014730afc7ed0add3 100644 --- a/python/paddle/fluid/tests/unittests/my_data_generator.py +++ b/python/paddle/fluid/tests/unittests/my_data_generator.py @@ -34,8 +34,5 @@ class MyDataset(fleet.MultiSlotDataGenerator): if __name__ == "__main__": - #start = time.clock() d = MyDataset() d.run_from_stdin() - #elapsed = (time.clock() - start) - #print("Time used:",elapsed) diff --git a/python/paddle/fluid/tests/unittests/test_data_generator.py b/python/paddle/fluid/tests/unittests/test_data_generator.py index 310913f695b34bf59bcaa57e0b1f8e6ef12d9429..221b9ba7683de164dfd002c7157f22d61d2ed6a5 100644 --- a/python/paddle/fluid/tests/unittests/test_data_generator.py +++ b/python/paddle/fluid/tests/unittests/test_data_generator.py @@ -15,6 +15,7 @@ import unittest import paddle.distributed.fleet as fleet import os import sys +import platform class MyMultiSlotDataGenerator(fleet.MultiSlotDataGenerator): @@ -109,9 +110,11 @@ class TestMultiSlotStringDataGenerator(unittest.TestCase): class TestMultiSlotStringDataGenerator_2(unittest.TestCase): - @unittest.skipIf(sys.platform == "win32", - "Windows does not support this pipe command") 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"