Update test with new IndexSwitchOp interface

ashermancinelli · ashermancinelli · commit 12dc35771c22 · 2025-11-11T08:02:14.000-08:00
diff --git a/mlir/test/python/ir/operation.py b/mlir/test/python/ir/operation.py
@@ -1213,7 +1213,7 @@ def testIndexSwitch():
             def index_switch(index):
                 c1 = arith.constant(i32, 1)
                 switch_op = scf.IndexSwitchOp(
-                    results_=[i32], arg=index, cases=range(3), num_caseRegions=3
+                    results_=[i32], arg=index, cases=range(3)
                 )
 
                 assert len(switch_op.regions) == 4

Original file line number	Diff line number	Diff line change
`@@ -1213,7 +1213,7 @@ def testIndexSwitch():`
`1213`	`1213`	`def index_switch(index):`
`1214`	`1214`	`c1 = arith.constant(i32, 1)`
`1215`	`1215`	`switch_op = scf.IndexSwitchOp(`
`1216`		`- results_=[i32], arg=index, cases=range(3), num_caseRegions=3`
	`1216`	`+ results_=[i32], arg=index, cases=range(3)`
`1217`	`1217`	`)`
`1218`	`1218`
`1219`	`1219`	`assert len(switch_op.regions) == 4`