diff --git a/python/paddle/v2/framework/tests/test_fill_zeros_like_op.py b/python/paddle/v2/framework/tests/test_fill_zeros_like_op.py index 69d89f537d67054dd74d8615d3c4e62f7285b13a..6e496f1be93cb1ff4e14b23687fa4e526f3c09ef 100644 --- a/python/paddle/v2/framework/tests/test_fill_zeros_like_op.py +++ b/python/paddle/v2/framework/tests/test_fill_zeros_like_op.py @@ -1,33 +1,15 @@ import unittest -import paddle.v2.framework.create_op_creation_methods as creation -import paddle.v2.framework.core as core +from op_test_util import OpTestMeta import numpy class TestFillZerosLikeOp(unittest.TestCase): - def test_fill(self): - scope = core.Scope() - a = scope.create_var("input") - a_tensor = a.get_tensor() - a_tensor.set_dims([546, 291]) - a_tensor.alloc_float() - a_tensor.set(numpy.random.random((546, 291)).astype("float32")) + __metaclass__ = OpTestMeta - op = creation.op_creations.fill_zeros_like(Src="input", Dst="output") - - for out in op.outputs(): - if scope.get_var(out) is None: - scope.create_var(out).get_tensor() - - b_tensor = scope.get_var("output").get_tensor() - op.infer_shape(scope) - self.assertEqual([546, 291], b_tensor.shape()) - ctx = core.DeviceContext.cpu_context() - op.run(scope, ctx) - b_tensor_array = numpy.array(b_tensor) - for r in range(0, 546): - for c in range(0, 291): - self.assertEqual(b_tensor_array[r][c], 0.0) + def setUp(self): + self.type = "fill_zeros_like" + self.Src = numpy.random.random((219, 232)).astype("float32") + self.Dst = numpy.zeros_like(self.Src) if __name__ == '__main__':