From 34735e96dde5d623ddde73b54a71ee9a55ddbfbd Mon Sep 17 00:00:00 2001 From: zhhsplendid Date: Mon, 21 Sep 2020 14:56:30 +0000 Subject: [PATCH] Fix unittest, test=develop --- .../dygraph_to_static/test_origin_info.py | 8 ++-- .../test_program_translator.py | 40 +++++++++---------- .../test_variable_trans_func.py | 10 ++--- 3 files changed, 27 insertions(+), 31 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_origin_info.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_origin_info.py index 3f77e9ade28..144b16873aa 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_origin_info.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_origin_info.py @@ -65,7 +65,7 @@ class TestOriginInfo(unittest.TestCase): self.func = simple_func def set_static_lineno(self): - self.static_abs_lineno_list = [2, 3, 4] + self.static_abs_lineno_list = [3, 4, 5] def set_dygraph_info(self): self.line_num = 3 @@ -149,7 +149,7 @@ class TestOriginInfoWithNestedFunc(TestOriginInfo): self.func = nested_func def set_static_lineno(self): - self.static_abs_lineno_list = [2, 4, 5, 6, 7] + self.static_abs_lineno_list = [3, 5, 6, 7, 8] def set_dygraph_info(self): self.line_num = 5 @@ -174,7 +174,7 @@ class TestOriginInfoWithDecoratedFunc(TestOriginInfo): self.func = decorated_func def set_static_lineno(self): - self.static_abs_lineno_list = [2, 3] + self.static_abs_lineno_list = [3, 4] def set_dygraph_info(self): self.line_num = 2 @@ -208,7 +208,7 @@ class TestOriginInfoWithDecoratedFunc2(TestOriginInfo): self.func = decorated_func2 def set_static_lineno(self): - self.static_abs_lineno_list = [2, 3] + self.static_abs_lineno_list = [3, 4] def set_dygraph_info(self): self.line_num = 2 diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_program_translator.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_program_translator.py index 873d9ecb535..d1134e40a38 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_program_translator.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_program_translator.py @@ -63,11 +63,9 @@ def get_source_code(func): class StaticCode1(): # TODO: Transform return statement def dyfunc_with_if_else(x_v, label=None): - __return_1 = fluid.layers.fill_constant( - shape=[1], dtype='bool', value=False) - __return_0 = fluid.layers.fill_constant( - shape=[1], dtype='bool', value=False) - __return_value_init_0 = fluid.layers.fill_constant( + __return_1 = paddle.fill_constant(shape=[1], dtype='bool', value=False) + __return_0 = paddle.fill_constant(shape=[1], dtype='bool', value=False) + __return_value_init_0 = paddle.fill_constant( shape=[1], dtype='float64', value=0.0) __return_value_0 = __return_value_init_0 @@ -79,13 +77,13 @@ class StaticCode1(): x_v = x_v + 1 return x_v - x_v = fluid.dygraph.dygraph_to_static.convert_operators.convert_ifelse( + x_v = paddle.jit.dygraph_to_static.convert_operators.convert_ifelse( fluid.layers.mean(x_v)[0] > 5, true_fn_0, false_fn_0, (x_v, ), (x_v, ), (x_v, )) def true_fn_1(__return_0, __return_value_0, label, x_v): loss = fluid.layers.cross_entropy(x_v, label) - __return_0 = fluid.layers.fill_constant( + __return_0 = paddle.fill_constant( shape=[1], dtype='bool', value=True) __return_value_0 = loss return __return_0, __return_value_0 @@ -94,13 +92,13 @@ class StaticCode1(): return __return_0, __return_value_0 __return_0, __return_value_0 = ( - fluid.dygraph.dygraph_to_static.convert_operators.convert_ifelse( + paddle.jit.dygraph_to_static.convert_operators.convert_ifelse( label is not None, true_fn_1, false_fn_1, (__return_0, __return_value_0, label, x_v), (__return_0, __return_value_0), (__return_0, __return_value_0))) def true_fn_2(__return_1, __return_value_0, x_v): - __return_1 = fluid.layers.fill_constant( + __return_1 = paddle.fill_constant( shape=[1], dtype='bool', value=True) __return_value_0 = x_v return __return_1, __return_value_0 @@ -109,8 +107,8 @@ class StaticCode1(): return __return_1, __return_value_0 __return_1, __return_value_0 = ( - fluid.dygraph.dygraph_to_static.convert_operators.convert_ifelse( - fluid.dygraph.dygraph_to_static.convert_operators. + paddle.jit.dygraph_to_static.convert_operators.convert_ifelse( + paddle.jit.dygraph_to_static.convert_operators. convert_logical_not(__return_0), true_fn_2, false_fn_2, (__return_1, __return_value_0, x_v), (__return_1, __return_value_0), (__return_1, __return_value_0))) @@ -120,11 +118,9 @@ class StaticCode1(): class StaticCode2(): # TODO: Transform return statement def dyfunc_with_if_else(x_v, label=None): - __return_3 = fluid.layers.fill_constant( - shape=[1], dtype='bool', value=False) - __return_2 = fluid.layers.fill_constant( - shape=[1], dtype='bool', value=False) - __return_value_init_1 = fluid.layers.fill_constant( + __return_3 = paddle.fill_constant(shape=[1], dtype='bool', value=False) + __return_2 = paddle.fill_constant(shape=[1], dtype='bool', value=False) + __return_value_init_1 = paddle.fill_constant( shape=[1], dtype='float64', value=0.0) __return_value_1 = __return_value_init_1 @@ -136,13 +132,13 @@ class StaticCode2(): x_v = x_v + 1 return x_v - x_v = fluid.dygraph.dygraph_to_static.convert_operators.convert_ifelse( + x_v = paddle.jit.dygraph_to_static.convert_operators.convert_ifelse( fluid.layers.mean(x_v)[0] > 5, true_fn_3, false_fn_3, (x_v, ), (x_v, ), (x_v, )) def true_fn_4(__return_2, __return_value_1, label, x_v): loss = fluid.layers.cross_entropy(x_v, label) - __return_2 = fluid.layers.fill_constant( + __return_2 = paddle.fill_constant( shape=[1], dtype='bool', value=True) __return_value_1 = loss return __return_2, __return_value_1 @@ -151,13 +147,13 @@ class StaticCode2(): return __return_2, __return_value_1 __return_2, __return_value_1 = ( - fluid.dygraph.dygraph_to_static.convert_operators.convert_ifelse( + paddle.jit.dygraph_to_static.convert_operators.convert_ifelse( label is not None, true_fn_4, false_fn_4, (__return_2, __return_value_1, label, x_v), (__return_2, __return_value_1), (__return_2, __return_value_1))) def true_fn_5(__return_3, __return_value_1, x_v): - __return_3 = fluid.layers.fill_constant( + __return_3 = paddle.fill_constant( shape=[1], dtype='bool', value=True) __return_value_1 = x_v return __return_3, __return_value_1 @@ -166,8 +162,8 @@ class StaticCode2(): return __return_3, __return_value_1 __return_3, __return_value_1 = ( - fluid.dygraph.dygraph_to_static.convert_operators.convert_ifelse( - fluid.dygraph.dygraph_to_static.convert_operators. + paddle.jit.dygraph_to_static.convert_operators.convert_ifelse( + paddle.jit.dygraph_to_static.convert_operators. convert_logical_not(__return_2), true_fn_5, false_fn_5, (__return_3, __return_value_1, x_v), (__return_3, __return_value_1), (__return_3, __return_value_1))) diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_variable_trans_func.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_variable_trans_func.py index e7632a0ba8d..e79209cb538 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_variable_trans_func.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_variable_trans_func.py @@ -51,24 +51,24 @@ class TestDataLayerNotCheck(unittest.TestCase): class TestVariableTransFunc(unittest.TestCase): def test_create_fill_constant_node(self): node = create_fill_constant_node("a", 1.0) - source = "a = fluid.layers.fill_constant(shape=[1], dtype='float64', value=1.0)" + source = "a = paddle.fill_constant(shape=[1], dtype='float64', value=1.0)" self.assertEqual(ast_to_source_code(node).strip(), source) node = create_fill_constant_node("b", True) - source = "b = fluid.layers.fill_constant(shape=[1], dtype='bool', value=True)" + source = "b = paddle.fill_constant(shape=[1], dtype='bool', value=True)" self.assertEqual(ast_to_source_code(node).strip(), source) if six.PY2: node = create_fill_constant_node("c", 214) - source = "c = fluid.layers.fill_constant(shape=[1], dtype='int32', value=214)" + source = "c = paddle.fill_constant(shape=[1], dtype='int32', value=214)" self.assertEqual(ast_to_source_code(node).strip(), source) node = create_fill_constant_node("d", long(10086)) - source = "d = fluid.layers.fill_constant(shape=[1], dtype='int64', value=10086)" + source = "d = paddle.fill_constant(shape=[1], dtype='int64', value=10086)" self.assertEqual(ast_to_source_code(node).strip(), source) else: node = create_fill_constant_node("c", 4293) - source = "c = fluid.layers.fill_constant(shape=[1], dtype='int64', value=4293)" + source = "c = paddle.fill_constant(shape=[1], dtype='int64', value=4293)" self.assertEqual(ast_to_source_code(node).strip(), source) self.assertIsNone(create_fill_constant_node("e", None)) -- GitLab