test_kmeans.py 文件源码

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