make assert_mode a parameter

Signed-off-by: martinvuyk <[email protected]>

stdlib/src/collections/_index_normalization.mojo

@@ -21,6 +21,7 @@ fn normalize_index[
     *,
     ignore_zero_length: Bool = False,
     cap_to_container_length: Bool = True,
+    assert_mode: StringLiteral = "warn",
 ](idx: Int, container: ContainerType) -> Int:
     """Normalize the given index value to a valid index value for the given
     container length. If the provided value is negative, the `index +
@@ -31,6 +32,7 @@ fn normalize_index[
         container_name: The name of the container. Used for the error message.
         ignore_zero_length: Whether to ignore if the container is of length 0.
         cap_to_container_length: Whether to cap the value to container length.
+        assert_mode: The mode in which to do the bounds check asserts.
 
     Args:
         idx: The index value to normalize.
@@ -49,13 +51,13 @@ fn normalize_index[
 
     @parameter
     if not ignore_zero_length:
-        debug_assert[assert_mode="none", cpu_only=True](
+        debug_assert[assert_mode=assert_mode, cpu_only=True](
             c_len > 0,
             "indexing into a ",
             container_name,
             " that has 0 elements",
         )
-        debug_assert[assert_mode="none", cpu_only=True](
+        debug_assert[assert_mode=assert_mode, cpu_only=True](
             -c_len <= idx < c_len,
             container_name,
             " has length: ",

stdlib/test/collections/test_index_normalization.mojo

@@ -20,13 +20,13 @@ from testing import assert_equal
 def test_out_of_bounds_message():
     l = List[Int](1, 2)
     # CHECK: index out of bounds: 2
-    _ = normalize_index["List"](2, l)
+    _ = normalize_index["List", assert_mode="safe"](2, l)
     # CHECK: index out of bounds: -3
-    _ = normalize_index["List"](-3, l)
+    _ = normalize_index["List", assert_mode="safe"](-3, l)
 
     l2 = List[Int]()
     # CHECK: indexing into a List that has 0 elements
-    _ = normalize_index["List"](2, l2)
+    _ = normalize_index["List", assert_mode="safe"](2, l2)
 
 
 def test_normalize_index():