Ensure that any/in/all work

Crunch-io · Jun 6, 2024 · feef5c2 · feef5c2
1 parent 70511b1
commit feef5c2
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/scrunch/tests/test_expressions.py b/scrunch/tests/test_expressions.py
@@ -795,6 +795,28 @@ def test_mr_any_subvar(self):
             ]
         }
 
+    def test_mr_all_subvar(self):
+        expr = "MyMrVar.all([subvar1, subvar2])"
+        expr_obj = parse_expr(expr)
+        assert expr_obj == {
+            'function': 'all',
+            'args': [
+                {'variable': 'MyMrVar'},
+                {'column': ['subvar1', 'subvar2']}
+            ]
+        }
+
+    def test_mr_in_subvar(self):
+        expr = "MyMrVar in [subvar1, subvar2]"
+        expr_obj = parse_expr(expr)
+        assert expr_obj == {
+            'function': 'in',
+            'args': [
+                {'variable': 'MyMrVar'},
+                {'column': ['subvar1', 'subvar2']}
+            ]
+        }
+
     def test_parse_sample_any(self):
         # 'text': 'CompanyTurnover is NA',
         # 'index_mapper': {'CompanyTurnover': any([99])}},