diff --git a/python/paddle/v2/framework/tests/test_gaussian_random_op.py b/python/paddle/v2/framework/tests/test_gaussian_random_op.py index 20c68007b5cd23caa57f7b2a81041c3549a5b521..f95ed70b58d611b3233a21d3f2a34c864ae4d1b3 100644 --- a/python/paddle/v2/framework/tests/test_gaussian_random_op.py +++ b/python/paddle/v2/framework/tests/test_gaussian_random_op.py @@ -6,12 +6,13 @@ import numpy class GaussianRandomTest(unittest.TestCase): def test_cpu(self): - self.test_gaussian_random(place=core.CPUPlace()) + self.gaussian_random_test(place=core.CPUPlace()) def test_gpu(self): - self.test_gaussian_random(place=core.GPUPlace(0)) + if core.is_compile_gpu(): + self.gaussian_random_test(place=core.GPUPlace(0)) - def test_gaussian_random(self, place): + def gaussian_random_test(self, place): scope = core.Scope() scope.new_var("Out").get_tensor() @@ -27,8 +28,8 @@ class GaussianRandomTest(unittest.TestCase): context = core.DeviceContext.create(place) op.run(scope, context) tensor = numpy.array(scope.find_var("Out").get_tensor()) - self.assertAlmostEqual(numpy.mean(tensor), .0, places=3) - self.assertAlmostEqual(numpy.std(tensor), 1., places=3) + self.assertAlmostEqual(numpy.mean(tensor), .0, delta=0.1) + self.assertAlmostEqual(numpy.std(tensor), 1., delta=0.1) if __name__ == '__main__':