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)
