Extend Operators

test/conftest.py

@@ -0,0 +1,33 @@
+import pytest
+from pysnark import runtime
+
+@pytest.fixture(scope="session")
+def disable_output():
+    """
+    Disables all circuit outputs when running tests
+    """
+    yield
+    runtime.autoprove = False
+
+def load_backend(backend):
+    import sys
+    import importlib
+    from pysnark import runtime
+
+    backends = [
+        ["libsnark",    "pysnark.libsnark.backend"],
+        ["libsnarkgg",  "pysnark.libsnark.backendgg"],
+        ["qaptools",    "pysnark.qaptools.backend"],
+        ["snarkjs",     "pysnark.snarkjsbackend"],
+        ["zkinterface", "pysnark.zkinterface.backend"],
+        ["zkifbellman", "pysnark.zkinterface.backendbellman"],
+        ["zkifbulletproofs", "pysnark.zkinterface.backendbulletproofs"],    
+        ["nobackend",   "pysnark.nobackend"]
+    ]
+
+    for mod in backends:
+        if backend == mod[0]:
+            runtime.backend = importlib.import_module(mod[1])
+            return
+
+    raise RuntimeError("Invalid backend for test")

test/test_branching.py

@@ -0,0 +1,31 @@
+import pytest
+from pysnark.runtime import PrivVal
+from pysnark.boolean import PrivValBool
+from pysnark.fixedpoint import PrivValFxp
+from pysnark.branching import if_then_else
+
+class TestLinCombFxp():
+    def test_lincomb_branch(self):
+        assert if_then_else(PrivValBool(0), PrivVal(1), PrivVal(2)).val() == 2
+        assert if_then_else(PrivValBool(1), PrivVal(1), PrivVal(2)).val() == 1
+
+    def test_lincombbool_branch(self):
+        assert if_then_else(PrivValBool(0), PrivValBool(1), PrivValBool(0)).val() == 0
+        assert if_then_else(PrivValBool(1), PrivValBool(1), PrivValBool(0)).val() == 1
+
+    def test_lincombfxp_branch(self):
+        assert if_then_else(PrivValBool(0), PrivValFxp(1.5), PrivValFxp(2.5)).val() == 2.5
+        assert if_then_else(PrivValBool(1), PrivValFxp(1.5), PrivValFxp(2.5)).val() == 1.5
+
+    def test_mixed_branch(self):
+        assert if_then_else(PrivValBool(0), PrivVal(1), PrivValFxp(2.5)).val() == 2.5
+        assert if_then_else(PrivValBool(1), PrivValFxp(1.5), PrivValBool(1)).val() == 1.5
+        assert if_then_else(PrivValBool(1), PrivValBool(1), PrivVal(2)).val() == 1
+
+    def test_branch_condition(self):
+        with pytest.raises(RuntimeError):
+            if_then_else(PrivVal(True), PrivVal(1), PrivVal(2))
+        with pytest.raises(RuntimeError):
+            if_then_else(PrivVal(1), PrivVal(1), PrivVal(2))
+        with pytest.raises(RuntimeError):
+            if_then_else(PrivValFxp(1), PrivVal(1), PrivVal(2))

test/test_lincomb.py

@@ -0,0 +1,148 @@
+from pysnark import runtime
+from pysnark.runtime import PrivVal, PubVal
+
+class TestLinComb():
+    def test_priv_val(self):
+        assert PrivVal(3).val() == 3
+
+    def test_pub_val(self):
+        assert PubVal(3).val() == 3
+
+    def test_add(self):
+        assert (PubVal(1) + 2).val() == 3
+        assert (2 + PubVal(1)).val() == 3
+        assert (PrivVal(1) + PrivVal(2)).val() == 3
+
+    def test_mul(self):
+        assert (PrivVal(2) * 2).val() == 4
+        assert (2 * PrivVal(2)).val() == 4
+        assert (PrivVal(2) * PrivVal(2)).val() == 4
+
+    def test_sub(self):
+        assert (PrivVal(2) - 1).val() == 1
+        assert (2 - PrivVal(1)).val() == 1
+        assert (PrivVal(2) - PrivVal(1)).val() == 1
+
+    def test_div_integer(self):
+        assert (PrivVal(4) / 2).val() == 2
+        assert (4 / PrivVal(2)).val() == 2
+        assert (PrivVal(4) / PrivVal(2)).val() == 2
+
+    def test_floor_div(self):
+        assert (PrivVal(4) // 2).val() == 2
+        assert (4 // PrivVal(2)).val() == 2
+        assert (PrivVal(4) // PrivVal(2)).val() == 2
+
+        assert (PrivVal(5) // 2).val() == 2
+        assert (5 // PrivVal(2)).val() == 2
+        assert (PrivVal(5) // PrivVal(2)).val() == 2
+
+    def test_mod(self):
+        assert (PrivVal(5) % 2).val() == 1
+        assert (5 % PrivVal(2)).val() == 1
+        assert (PrivVal(5) % PrivVal(2)).val() == 1
+
+        assert (PrivVal(5) % 1).val() == 0
+        assert (5 % PrivVal(1)).val() == 0
+        assert (PrivVal(5) % PrivVal(1)).val() == 0
+
+    def test_pow(self):
+        runtime.bitlength = 5
+
+        assert (PrivVal(2) ** 4).val() == 16
+        assert (2 ** PrivVal(4)).val() == 16
+        assert (PrivVal(2) ** PrivVal(4)).val() == 16
+
+        assert (PrivVal(2) ** 0).val() == 1
+        assert (2 ** PrivVal(0)).val() == 1
+        assert (PrivVal(2) ** PrivVal(0)).val() == 1
+
+        runtime.bitlength = 16
+
+    def test_rshift(self):
+        runtime.bitlength = 5
+
+        assert (PrivVal(2) << 2).val() == 8
+        assert (2 << PrivVal(2)).val() == 8
+        assert (PrivVal(2) << PrivVal(2)).val() == 8
+
+        runtime.bitlength = 16
+
+    def test_lshift(self):
+        runtime.bitlength = 5
+
+        assert (PrivVal(8) >> 2).val() == 2
+        assert (8 >> PrivVal(2)).val() == 2
+        assert (PrivVal(8) >> PrivVal(2)).val() == 2
+
+        runtime.bitlength = 16
+
+    def test_lt(self):
+        assert((PrivVal(1) < 2).val() == 1)
+        assert((1 < PrivVal(2)).val() == 1)
+        assert((PrivVal(1) < PrivVal(2)).val() == 1)
+
+        assert((PrivVal(2) < 1).val() == 0)
+        assert((2 < PrivVal(1)).val() == 0)
+        assert((PrivVal(2) < PrivVal(1)).val() == 0)
+
+        assert((PrivVal(1) < 1).val() == 0)
+        assert((1 < PrivVal(1)).val() == 0)
+        assert((PrivVal(1) < PrivVal(1)).val() == 0)
+
+    def test_gt(self):
+        assert((PrivVal(2) > 1).val() == 1)
+        assert((2 > PrivVal(1)).val() == 1)
+        assert((PrivVal(2) > PrivVal(1)).val() == 1)
+
+        assert((PrivVal(1) > 2).val() == 0)
+        assert((1 > PrivVal(2)).val() == 0)
+        assert((PrivVal(1) > PrivVal(2)).val() == 0)
+
+        assert((PrivVal(1) > 1).val() == 0)
+        assert((1 > PrivVal(1)).val() == 0)
+        assert((PrivVal(1) > PrivVal(1)).val() == 0)
+
+    def test_lte(self):
+        assert((PrivVal(1) <= 2).val() == 1)
+        assert((1 <= PrivVal(2)).val() == 1)
+        assert((PrivVal(1) <= PrivVal(2)).val() == 1)
+
+        assert((PrivVal(2) <= 1).val() == 0)
+        assert((2 <= PrivVal(1)).val() == 0)
+        assert((PrivVal(2) <= PrivVal(1)).val() == 0)
+
+        assert((PrivVal(1) <= 1).val() == 1)
+        assert((1 <= PrivVal(1)).val() == 1)
+        assert((PrivVal(1) <= PrivVal(1)).val() == 1)
+
+    def test_gte(self):
+        assert((PrivVal(2) >= 1).val() == 1)
+        assert((2 >= PrivVal(1)).val() == 1)
+        assert((PrivVal(2) >= PrivVal(1)).val() == 1)
+
+        assert((PrivVal(1) >= 2).val() == 0)
+        assert((1 >= PrivVal(2)).val() == 0)
+        assert((PrivVal(1) >= PrivVal(2)).val() == 0)
+
+        assert((PrivVal(1) >= 1).val() == 1)
+        assert((1 >= PrivVal(1)).val() == 1)
+        assert((PrivVal(1) >= PrivVal(1)).val() == 1)
+
+    def test_eq(self):
+        assert((PrivVal(2) == 2).val() == 1)
+        assert((2 == PrivVal(2)).val() == 1)
+        assert((PrivVal(2) == PrivVal(2)).val() == 1)
+
+        assert((PrivVal(2) == 1).val() == 0)
+        assert((2 == PrivVal(1)).val() == 0)
+        assert((PrivVal(2) == PrivVal(1)).val() == 0)
+
+    def test_neq(self):
+        assert((PrivVal(2) != 1).val() == 1)
+        assert((2 != PrivVal(1)).val() == 1)
+        assert((PrivVal(2) != PrivVal(1)).val() == 1)
+
+        assert((PrivVal(2) != 2).val() == 0)
+        assert((2 != PrivVal(2)).val() == 0)
+        assert((PrivVal(2) != PrivVal(2)).val() == 0)

test/test_lincomb_bitwise.py

@@ -0,0 +1,33 @@
+import pytest
+from pysnark.runtime import PrivVal
+from pysnark.boolean import LinCombBool
+
+class TestLinCombBitwise():
+    def test_to_bits(self):
+        bits = PrivVal(5).to_bits()
+        assert bits[3].val() == 0
+        assert bits[2].val() == 1
+        assert bits[1].val() == 0
+        assert bits[0].val() == 1
+
+    def test_and(self):
+        assert (3 & PrivVal(7)).val() == 3
+        assert (PrivVal(7) & 3).val() == 3
+        assert (PrivVal(7) & PrivVal(3)).val() == 3
+        assert (PrivVal(7) & PrivVal(0)).val() == 0
+        assert (PrivVal(0) & PrivVal(7)).val() == 0
+
+    def test_or(self):
+        assert (3 | PrivVal(7)).val() == 7
+        assert (PrivVal(7) | 3).val() == 7
+        assert (PrivVal(7) | PrivVal(3)).val() == 7
+        assert (PrivVal(7) | PrivVal(0)).val() == 7
+        assert (PrivVal(0) | PrivVal(7)).val() == 7
+
+    def test_xor(self):
+        assert (3 ^ PrivVal(7)).val() == 4
+        assert (PrivVal(7) ^ 3).val() == 4
+        assert (PrivVal(7) ^ PrivVal(3)).val() == 4
+        assert (PrivVal(7) ^ PrivVal(0)).val() == 7
+        assert (PrivVal(0) ^ PrivVal(7)).val() == 7
+