From 16c5f629bdf162d7cdf81150bb5e3af2a1a087c0 Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Mon, 25 Sep 2017 16:46:35 -0700 Subject: [PATCH] Complete unittest for OP --- python/paddle/v2/framework/tests/test_protobuf_descs.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/v2/framework/tests/test_protobuf_descs.py b/python/paddle/v2/framework/tests/test_protobuf_descs.py index aa9a0b33ac6..14367e5d594 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): -- GitLab