op_test_util.py 2.5 KB
Newer Older
Y
Yu Yang 已提交
1 2 3 4 5 6 7
import paddle.v2.framework.core as core
import unittest
import numpy
import paddle.v2.framework.create_op_creation_methods as creation


class OpTestMeta(type):
Y
Yu Yang 已提交
8 9 10 11 12 13 14 15 16 17 18 19
    """
    Operator Test ClassMeta.
    
    It injects `test_all` method into user's OperatorTest class, to make Python 
    unittest module run that method.
    
    The `test_all` read what value is stored in `self`. It use self's values to
    create and run a operator, and check whether that op is OK or not.
    
    See `test_add_two_op` for example usage.
    """

Y
Yu Yang 已提交
20 21 22 23 24 25 26
    def __new__(cls, name, bases, attrs):
        obj = super(OpTestMeta, cls).__new__(cls, name, bases, attrs)

        def test_all(self):
            func = getattr(creation.op_creations, self.type, None)
            self.assertIsNotNone(func)

Y
Yu Yang 已提交
27
            scope = core.Scope()
Y
Yu Yang 已提交
28
            kwargs = dict()
Q
qijun 已提交
29 30 31 32 33 34 35 36 37
            places = []
            places.append(core.CPUPlace())
            if core.is_compile_gpu():
                places.append(core.GPUPlace(0))

            for place in places:
                for in_name in func.all_input_args:
                    if hasattr(self, in_name):
                        kwargs[in_name] = in_name
Q
qijun 已提交
38
                        var = scope.new_var(in_name).get_tensor()
Q
qijun 已提交
39 40 41 42 43
                        arr = getattr(self, in_name)
                        var.set_dims(arr.shape)
                        var.set(arr, place)
                    else:
                        kwargs[in_name] = "@EMPTY@"
Y
Yu Yang 已提交
44

Q
qijun 已提交
45 46 47
                for out_name in func.all_output_args:
                    if hasattr(self, out_name):
                        kwargs[out_name] = out_name
Q
qijun 已提交
48
                        scope.new_var(out_name).get_tensor()
Y
Yu Yang 已提交
49

Q
qijun 已提交
50 51 52
                for attr_name in func.all_attr_args:
                    if hasattr(self, attr_name):
                        kwargs[attr_name] = getattr(self, attr_name)
Y
Yu Yang 已提交
53

Q
qijun 已提交
54
                op = func(**kwargs)
Y
Yu Yang 已提交
55

Q
qijun 已提交
56
                op.infer_shape(scope)
Y
Yu Yang 已提交
57

Q
qijun 已提交
58 59
                ctx = core.DeviceContext.create(place)
                op.run(scope, ctx)
Y
Yu Yang 已提交
60

Q
qijun 已提交
61
                for out_name in func.all_output_args:
Q
qijun 已提交
62
                    actual = numpy.array(scope.find_var(out_name).get_tensor())
Q
qijun 已提交
63 64 65 66 67
                    expect = getattr(self, out_name)
                    # TODO(qijun) The default decimal is 7, but numpy.dot and eigen.mul
                    # has some diff, and could not pass unittest. So I set decimal 3 here.
                    # And I will check this in future.
                    numpy.testing.assert_almost_equal(actual, expect, decimal=3)
Y
Yu Yang 已提交
68 69 70

        obj.test_all = test_all
        return obj