From 4c70e3fe0b19b911ba7c6276ec2aa592036b64f9 Mon Sep 17 00:00:00 2001 From: Jesse Bannon Date: Tue, 21 Nov 2023 17:50:58 -0800 Subject: [PATCH] test nested lambdas --- tests/unit/script/types/test_function.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/unit/script/types/test_function.py b/tests/unit/script/types/test_function.py index 54a8edce..4e5bf55e 100644 --- a/tests/unit/script/types/test_function.py +++ b/tests/unit/script/types/test_function.py @@ -149,3 +149,14 @@ class TestFunction: "power_of_4": "{%power_4(2)}", } ).resolve() == {"power_of_4": Integer(16)} + + def test_nested_lambda_custom_functions_within_custom_functions(self): + assert Script( + { + "%nest4": "{%mul($0, 2)}", + "%nest3": "{%array_at(%array_apply([$0], %nest4), 0)}", + "%nest2": "{%array_at(%array_apply([$0], %nest3), 0)}", + "%nest1": "{%array_at(%array_apply([$0], %nest2), 0)}", + "output": "{%array_at(%array_apply([2], %nest1), 0)}", + } + ).resolve() == {"output": Integer(4)}