From c59fe7a039c1b6b9334094cfd151d79bf86a37c1 Mon Sep 17 00:00:00 2001
From: fengjiayi <fengjiayi@baidu.com>
Date: Tue, 1 Aug 2017 10:47:27 -0700
Subject: [PATCH] Add unittest for FillZerosLikeOp

---
 .../tests/test_fill_zeros_like_op.py          | 35 +++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 python/paddle/v2/framework/tests/test_fill_zeros_like_op.py

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
new file mode 100644
index 000000000..772edf895
--- /dev/null
+++ b/python/paddle/v2/framework/tests/test_fill_zeros_like_op.py
@@ -0,0 +1,35 @@
+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(None)
+        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"))
+
+        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)
+
+
+if __name__ == '__main__':
+    unittest.main()
-- 
GitLab