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

Rewirte test based on python test frameworks

上级 244c343b
import unittest import unittest
import paddle.v2.framework.create_op_creation_methods as creation from op_test_util import OpTestMeta
import paddle.v2.framework.core as core
import numpy import numpy
class TestFillZerosLikeOp(unittest.TestCase): class TestFillZerosLikeOp(unittest.TestCase):
def test_fill(self): __metaclass__ = OpTestMeta
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"))
op = creation.op_creations.fill_zeros_like(Src="input", Dst="output") def setUp(self):
self.type = "fill_zeros_like"
for out in op.outputs(): self.Src = numpy.random.random((219, 232)).astype("float32")
if scope.get_var(out) is None: self.Dst = numpy.zeros_like(self.Src)
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__': if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册