Fix mispelled function name.

PiperOrigin-RevId: 231429189
This commit is contained in:
Dan Moldovan 2019-01-29 10:36:32 -08:00 committed by TensorFlower Gardener
parent 5b16e7079a
commit 603e11f5e7
2 changed files with 11 additions and 1 deletions

View File

@ -202,7 +202,7 @@ class ReturnStatementsTransformer(converter.Base):
def visit_Return(self, node): def visit_Return(self, node):
self.state[_Return].used = True self.state[_Return].used = True
retval = node.value if node.value else parser.parse_expresison('None') retval = node.value if node.value else parser.parse_expression('None')
template = """ template = """
do_return_var_name = True do_return_var_name = True

View File

@ -187,6 +187,16 @@ class SingleReturnTest(converter_testing.TestCase):
self.assertTransformedEquivalent(test_fn, 2) self.assertTransformedEquivalent(test_fn, 2)
self.assertTransformedEquivalent(test_fn, 4) self.assertTransformedEquivalent(test_fn, 4)
def test_null_return(self):
def test_fn(n):
if n > 4:
return
return
self.assertTransformedEquivalent(test_fn, 4)
self.assertTransformedEquivalent(test_fn, 5)
if __name__ == '__main__': if __name__ == '__main__':
test.main() test.main()