From 99d533d026188925186d1ab188130f73897dca70 Mon Sep 17 00:00:00 2001
From: Qiao Longfei <qiaolongfei@baidu.com>
Date: Tue, 22 Jan 2019 13:36:30 +0800
Subject: [PATCH] update TestNumpyArrayInitializer test=develop

---
 python/paddle/fluid/initializer.py                      | 2 +-
 python/paddle/fluid/tests/unittests/test_initializer.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/python/paddle/fluid/initializer.py b/python/paddle/fluid/initializer.py
index 5e99007031..4f434328e4 100644
--- a/python/paddle/fluid/initializer.py
+++ b/python/paddle/fluid/initializer.py
@@ -734,7 +734,7 @@ class NumpyArrayInitializer(Initializer):
             outputs={'Out': var},
             attrs={
                 'dtype': dtype,
-                'shape': list(input.shape),
+                'shape': list(self._value.shape),
                 value_name: values
             },
             stop_gradient=True)
diff --git a/python/paddle/fluid/tests/unittests/test_initializer.py b/python/paddle/fluid/tests/unittests/test_initializer.py
index 2e70175d43..2d98b063d1 100644
--- a/python/paddle/fluid/tests/unittests/test_initializer.py
+++ b/python/paddle/fluid/tests/unittests/test_initializer.py
@@ -427,8 +427,8 @@ class TestNumpyArrayInitializer(unittest.TestCase):
         import numpy
         program = framework.Program()
         block = program.global_block()
+        np_array = numpy.random.random((10000)).astype("float32")
         for _ in range(2):
-            np_array = numpy.array([1, 2, 3, 4]).astype('float32')
             block.create_parameter(
                 dtype=np_array.dtype,
                 shape=np_array.shape,
@@ -438,6 +438,7 @@ class TestNumpyArrayInitializer(unittest.TestCase):
         self.assertEqual(len(block.ops), 1)
         init_op = block.ops[0]
         self.assertEqual(init_op.type, 'assign_value')
+        assert (init_op.attr('fp32_values') == np_array).all()
 
 
 if __name__ == '__main__':
-- 
GitLab