test_fill_zeros_like_op.py 412 字节
Newer Older
F
fengjiayi 已提交
1
import unittest
Q
qijun 已提交
2 3
import numpy as np
from op_test import OpTest
F
fengjiayi 已提交
4 5


Q
qijun 已提交
6
class TestFillZerosLikeOp(OpTest):
7
    def setUp(self):
Q
qijun 已提交
8
        self.op_type = "fill_zeros_like"
9
        self.inputs = {'X': np.random.random((219, 232)).astype("float32")}
F
fengjiayi 已提交
10
        self.outputs = {'Out': np.zeros_like(self.inputs["X"])}
Q
qijun 已提交
11 12 13

    def test_check_output(self):
        self.check_output()
F
fengjiayi 已提交
14 15


Q
qijun 已提交
16
if __name__ == "__main__":
F
fengjiayi 已提交
17
    unittest.main()