Merge pull request #30 from NillionNetwork/feat/nada-test

Add testing examples for addition program using nada-test

nada-project.toml

@@ -2,6 +2,9 @@ name = "nada-by-example"
 version = "0.1.0"
 authors = [""]
 
+[test_framework.nada-test]
+command = "nada-test ./tests"
+
 [[programs]]
 path = "src/main.py"
 prime_size = 128

requirements.txt

@@ -1,5 +1,6 @@
 nada-dsl
 nada-numpy
+nada-test
 
 # streamlit demo dependencies
 pyyaml

tests/addition_test.py

@@ -0,0 +1,20 @@
+# These tests use the nada-test testing framework
+from nada_test import nada_test, NadaTest
+
+# Functional style addition test
+@nada_test(program="addition")
+def addition_test_functional():
+    num_1 = 30
+    num_2 = 10
+    inputs = {"num_1": num_1, "num_2": num_2}
+    outputs = yield inputs
+    assert outputs["sum"] == num_1 + num_2
+
+# Class style addition test
+@nada_test(program="addition")
+class Test(NadaTest):
+    def inputs(self):
+        return {"num_1": 30, "num_2": 10}
+
+    def check(self, outputs):
+        assert outputs["sum"] == 40