test: add list folding lexer and parser tests

0xPolygonMiden · Jan 31, 2023 · 0d3b94e · 0d3b94e
1 parent 08763f9
commit 0d3b94e
Show file tree

Hide file tree

Showing 4 changed files with 592 additions and 3 deletions.
diff --git a/air-script-core/src/expression.rs b/air-script-core/src/expression.rs
@@ -28,7 +28,6 @@ pub enum Expression {
 
 #[derive(Debug, Clone, Eq, PartialEq)]
 pub enum ListFoldingType {
-    Expr(ListComprehension),
     Sum(ListComprehension),
     Prod(ListComprehension),
 }
diff --git a/ir/src/constraints/graph.rs b/ir/src/constraints/graph.rs
@@ -161,6 +161,7 @@ impl AlgebraicGraph {
                     lhs.domain(),
                 ))
             }
+            Expression::ListFolding(_) => todo!(),
         }
     }
 
@@ -228,7 +229,6 @@ impl AlgebraicGraph {
                 let lhs_base = self.accumulate_degree(cycles, lhs);
                 lhs_base * rhs
             }
-            Expression::ListFolding(_) => todo!(),
         }
     }
 

diff --git a/parser/src/lexer/tests/list_comprehension.rs b/parser/src/lexer/tests/list_comprehension.rs
@@ -72,3 +72,79 @@ fn multiple_iterables_comprehension() {
     ];
     expect_valid_tokenization(source, tokens);
 }
+
+#[test]
+fn one_iterable_folding() {
+    let source = "let y = sum([x for x in x])";
+    let tokens = vec![
+        Token::Let,
+        Token::Ident("y".to_string()),
+        Token::Equal,
+        Token::Sum,
+        Token::Lparen,
+        Token::Lsqb,
+        Token::Ident("x".to_string()),
+        Token::For,
+        Token::Ident("x".to_string()),
+        Token::In,
+        Token::Ident("x".to_string()),
+        Token::Rsqb,
+        Token::Rparen,
+    ];
+    expect_valid_tokenization(source, tokens);
+}
+
+#[test]
+fn multiple_iterables_list_folding() {
+    let source = "let a = sum([w + x - y - z for (w, x, y, z) in (0..3, x, y[0..3], z[0..3])])";
+    let tokens = vec![
+        Token::Let,
+        Token::Ident("a".to_string()),
+        Token::Equal,
+        Token::Sum,
+        Token::Lparen,
+        Token::Lsqb,
+        Token::Ident("w".to_string()),
+        Token::Plus,
+        Token::Ident("x".to_string()),
+        Token::Minus,
+        Token::Ident("y".to_string()),
+        Token::Minus,
+        Token::Ident("z".to_string()),
+        Token::For,
+        Token::Lparen,
+        Token::Ident("w".to_string()),
+        Token::Comma,
+        Token::Ident("x".to_string()),
+        Token::Comma,
+        Token::Ident("y".to_string()),
+        Token::Comma,
+        Token::Ident("z".to_string()),
+        Token::Rparen,
+        Token::In,
+        Token::Lparen,
+        Token::Num("0".to_string()),
+        Token::Range,
+        Token::Num("3".to_string()),
+        Token::Comma,
+        Token::Ident("x".to_string()),
+        Token::Comma,
+        Token::Ident("y".to_string()),
+        Token::Lsqb,
+        Token::Num("0".to_string()),
+        Token::Range,
+        Token::Num("3".to_string()),
+        Token::Rsqb,
+        Token::Comma,
+        Token::Ident("z".to_string()),
+        Token::Lsqb,
+        Token::Num("0".to_string()),
+        Token::Range,
+        Token::Num("3".to_string()),
+        Token::Rsqb,
+        Token::Rparen,
+        Token::Rsqb,
+        Token::Rparen,
+    ];
+    expect_valid_tokenization(source, tokens);
+}