fix: Add slot range validation to range expressions which specify a step

algorandfoundation · Jan 25, 2024 · 3ccd47f · 3ccd47f
1 parent ad09eb8
commit 3ccd47f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/puya/awst_build/module.py b/src/puya/awst_build/module.py
@@ -622,7 +622,10 @@ def map_urange_args(args: list[mypy.nodes.Expression]) -> range:
                 mypy.nodes.IntExpr(value=stop),
                 mypy.nodes.IntExpr(value=step),
             ]:
-                return range(start, stop, step)
+                the_range = range(start, stop, step)
+                for slot in the_range:
+                    check_slot_is_in_range(slot)
+                return the_range
             case _:
                 raise CodeError("Unexpected arguments for urange", source_location)