From 825fcd39b23e04498df4cc38d9a25295e9936d93 Mon Sep 17 00:00:00 2001 From: yaoxuefeng6 Date: Tue, 22 Sep 2020 11:03:01 +0800 Subject: [PATCH] fix ut --- python/paddle/fluid/tests/unittests/my_data_generator.py | 3 --- python/paddle/fluid/tests/unittests/test_data_generator.py | 7 +++++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/my_data_generator.py b/python/paddle/fluid/tests/unittests/my_data_generator.py index 6d5061edf6b..ac906b32561 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 310913f695b..221b9ba7683 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" -- GitLab