test_add_op.py 481 字节
Newer Older
Y
Yu Yang 已提交
1
import unittest
Q
qijun 已提交
2 3
import numpy as np
from op_test import OpTest
Q
Qiao Longfei 已提交
4

Y
Yu Yang 已提交
5

Q
qijun 已提交
6
class TestAddOp(OpTest):
Y
Yu Yang 已提交
7
    def setUp(self):
Q
qijun 已提交
8
        self.op_type = "add"
D
dangqingqing 已提交
9
        self.inputs = {
Q
qijun 已提交
10 11
            'X': np.random.random((102, 105)).astype("float32"),
            'Y': np.random.random((102, 105)).astype("float32")
D
dangqingqing 已提交
12 13
        }
        self.outputs = {'Out': self.inputs['X'] + self.inputs['Y']}
Y
Yu Yang 已提交
14

Q
qijun 已提交
15 16 17
    def test_check_output(self):
        self.check_output()

Y
Yu Yang 已提交
18

Q
qijun 已提交
19
if __name__ == "__main__":
Y
Yu Yang 已提交
20
    unittest.main()