add test for parsing Program

edin-dal · Jun 19, 2024 · c894330 · c894330
1 parent 6186e80
commit c894330
Showing 1 changed file with 71 additions and 0 deletions.
diff --git a/src/test/scala/uk/ac/ed/dal/structtensor/Parser/ParserTest.scala b/src/test/scala/uk/ac/ed/dal/structtensor/Parser/ParserTest.scala
@@ -204,4 +204,75 @@ class ParserTest extends AnyFlatSpec with Matchers with ParallelTestExecution {
       )
     )
   }
+
+  it should "parse a program" in {
+    val input =
+      """|foo(x, y, z):= bar(x, y, z) * baz(x, y, z) + qux(x, y, z)
+        |bax(x, y, z):= foo(x, y, z) * bar(x, y, z) + qux(x, y, z)
+        |""".stripMargin
+    val result = fastparse.parse(input, Parser.program(_))
+    result.isSuccess shouldBe true
+    result.get.value shouldBe Seq(
+      Rule(
+        Access("foo", Seq(Variable("x"), Variable("y"), Variable("z")), Tensor),
+        SoP(
+          Seq(
+            Prod(
+              Seq(
+                Access(
+                  "bar",
+                  Seq(Variable("x"), Variable("y"), Variable("z")),
+                  Tensor
+                ),
+                Access(
+                  "baz",
+                  Seq(Variable("x"), Variable("y"), Variable("z")),
+                  Tensor
+                )
+              )
+            ),
+            Prod(
+              Seq(
+                Access(
+                  "qux",
+                  Seq(Variable("x"), Variable("y"), Variable("z")),
+                  Tensor
+                )
+              )
+            )
+          )
+        )
+      ),
+      Rule(
+        Access("bax", Seq(Variable("x"), Variable("y"), Variable("z")), Tensor),
+        SoP(
+          Seq(
+            Prod(
+              Seq(
+                Access(
+                  "foo",
+                  Seq(Variable("x"), Variable("y"), Variable("z")),
+                  Tensor
+                ),
+                Access(
+                  "bar",
+                  Seq(Variable("x"), Variable("y"), Variable("z")),
+                  Tensor
+                )
+              )
+            ),
+            Prod(
+              Seq(
+                Access(
+                  "qux",
+                  Seq(Variable("x"), Variable("y"), Variable("z")),
+                  Tensor
+                )
+              )
+            )
+          )
+        )
+      )
+    )
+  }
 }