def test_program_translator(self): answer = "\ def dyfunc_with_if_else(x_v, label=None):\n\ \n\ def true_fn_1(x_v):\n\ x_v = x_v - 1\n\ return x_v\n\ \n\ def false_fn_1(x_v):\n\ x_v = x_v + 1\n\ return x_v\n\ x_v = fluid.layers.cond(fluid.layers.mean(x_v)[0] > 5, lambda :\n\ true_fn_1(x_v), lambda : false_fn_1(x_v))\n\ if label is not None:\n\ loss = fluid.layers.cross_entropy(x_v, label)\n\ return loss\n\ return x_v\n" program_translator = ProgramTranslator() code = program_translator.get_code(dyfunc_with_if_else) self.assertEqual(answer, code)
def test_program_translator(self): answer = get_source_code(StaticCode2.dyfunc_with_if_else) program_translator = ProgramTranslator() code = program_translator.get_code(dyfunc_with_if_else) self.assertEqual(answer, code)
def test_decorator(self): x_v = None program_translator = ProgramTranslator() code = program_translator.get_code(dyfunc_with_if_else) answer = get_source_code(StaticCode1.dyfunc_with_if_else) self.assertEqual(answer, code)
def test_code(self): answer = get_source_code(StaticCode.dyfunc_assign) program_translator = ProgramTranslator() code = program_translator.get_code(dyfunc_assign) self.assertEqual(answer, code)