Complete unittest for OP

8 years ago · 16c5f629bd
parent f9f910a33b
commit 16c5f629bd
1 changed files with 1 additions and 0 deletions
--- 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):