Add support for for loops in python interpreter (#24429)

Add support for for loops
This commit is contained in:
Sylvain Gugger
2023-06-26 09:58:14 -04:00
committed by GitHub
parent c2aa5e17e4
commit 5757923888
3 changed files with 23 additions and 0 deletions

View File

@@ -122,3 +122,10 @@ class PythonInterpreterTester(unittest.TestCase):
result = evaluate(code, {"add_two": add_two}, state=state)
assert result == 5
self.assertDictEqual(state, {"x": 3, "test_dict": {"x": 3, "y": 5}})
def test_evaluate_for(self):
code = "x = 0\nfor i in range(3):\n x = i"
state = {}
result = evaluate(code, {"range": range}, state=state)
assert result == 2
self.assertDictEqual(state, {"x": 2, "i": 2})