From eb5c33cb8ca4c7dce87d8a229ad4b11774133a95 Mon Sep 17 00:00:00 2001 From: silingtong123 <35439432+silingtong123@users.noreply.github.com> Date: Sat, 12 Oct 2019 10:24:41 +0800 Subject: [PATCH] Add uniform random unittest (#20485) * test=develop, add unittest of seed failed work * test=develop, add unittest of seed failed work --- .../tests/unittests/test_uniform_random_op.py | 26 +++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/python/paddle/fluid/tests/unittests/test_uniform_random_op.py b/python/paddle/fluid/tests/unittests/test_uniform_random_op.py index ca18a07753..92ac6be3c2 100644 --- a/python/paddle/fluid/tests/unittests/test_uniform_random_op.py +++ b/python/paddle/fluid/tests/unittests/test_uniform_random_op.py @@ -243,6 +243,32 @@ class TestUniformRandomOp_attr_tensor_API(unittest.TestCase): outs = exe.run(train_program, fetch_list=[ret]) +class TestUniformRandomOp_API_seed(unittest.TestCase): + def test_attr_tensor_API(self): + startup_program = fluid.Program() + train_program = fluid.Program() + with fluid.program_guard(train_program, startup_program): + _min = 5 + _max = 10 + _seed = 10 + ret = fluid.layers.nn.uniform_random( + [2, 3, 2], min=_min, max=_max, seed=_seed) + ret_2 = fluid.layers.nn.uniform_random( + [2, 3, 2], min=_min, max=_max, seed=_seed) + res = fluid.layers.equal(ret, ret_2) + place = fluid.CPUPlace() + if fluid.core.is_compiled_with_cuda(): + place = fluid.CUDAPlace(0) + exe = fluid.Executor(place) + + exe.run(startup_program) + ret_value, cmp_value = exe.run(train_program, fetch_list=[ret, res]) + self.assertTrue(np.array(cmp_value).all()) + for i in ret_value.flatten(): + self.assertGreaterEqual(i, _min) + self.assertLess(i, _max) + + class TestUniformRandomOpSelectedRowsShapeTensor(unittest.TestCase): def get_places(self): places = [core.CPUPlace()] -- GitLab