diff --git a/python/paddle/v2/framework/tests/test_protobuf_descs.py b/python/paddle/v2/framework/tests/test_protobuf_descs.py
index aa9a0b33ac6ad7e37202a79fb45baddf3d889338..14367e5d594f26062ef94200949e9a7a7f72d38b 100644
--- 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):