def test_python(self): self.assertEqual(self.evaluate(_basic_expr(True)), 1) self.assertEqual(self.evaluate(_basic_expr(False)), 2) self.assertEqual( conditional_expressions.if_exp(True, lambda: 1, lambda: 2, ''), 1) self.assertEqual( conditional_expressions.if_exp(False, lambda: 1, lambda: 2, ''), 2)
def test_fn(): conditional_expressions.if_exp( constant_op.constant(True), lambda: 1.0, lambda: 2, 'expr_repr')
def _basic_expr(cond): return conditional_expressions.if_exp( cond, lambda: constant_op.constant(1), lambda: constant_op.constant(2), 'cond')