fix unit test

2024-01-06 23:30:15 -08:00 · 2024-01-06 23:30:15 -08:00 · ad5a7b7fc5
commit ad5a7b7fc5
parent ac00b482e7
1 changed files with 5 additions and 1 deletions
--- a/tests/unit/script/types/test_function.py
+++ b/tests/unit/script/types/test_function.py
@ -49,7 +49,11 @@ class TestFunction:
    @pytest.mark.parametrize(
        "function_str, expected_types, received_types",
        [
-            ("{%array_at({'a': 'dict?'}, 1)}", "array: Array, idx: Integer", "Map, Integer"),
+            (
+                "{%array_at({'a': 'dict?'}, 1)}",
+                "array: Array, idx: Integer, default: Optional[AnyArgument]",
+                "Map, Integer",
+            ),
            ("{%array_extend('not', 'array')}", "arrays: Array, ...", "String, String"),
            (
                "{%replace('hi mom', 'mom', 'dad', 1, 0)}",