提交 589c3d86 编写于 作者: F fengjiayi

Rewirte test based on python test frameworks

上级 244c343b
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__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册