test_kmeans.py 文件源码

def test_default(self):
        output = example_test.run_example(
            'kmeans/kmeans.py', '-m', '1', '--num', '10')
        six.assertRegex(
            self, output.decode('utf-8'),
            r' CPU :  [0-9\.]+ sec\s+GPU :  [0-9\.]+ sec')