Complete unittest for OP

16c5f629 · Yu Yang · f9f910a3 · 16c5f629
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/v2/framework/tests/test_protobuf_descs.py python/paddle/v2/framework/tests/test_protobuf_descs.py +1 -0

未找到文件。
--- a/python/paddle/v2/framework/tests/test_protobuf_descs.py
+++ b/python/paddle/v2/framework/tests/test_protobuf_descs.py
@@ -53,6 +53,7 @@ class TestOpDesc(unittest.TestCase):

        op.set_block_attr("block_attr", prog.block(0))
        self.assertEqual(0, op.get_block_attr("block_attr"))
+        self.assertEqual(core.AttrType.INT, op.attr_type("int_attr"))


 class TestProgramDesc(unittest.TestCase):