walsh transform

tests/test_walsh.py

@@ -0,0 +1,11 @@
+from walsh.walsh import walsh_transform
+
+def test_walsh_xor_two_variables():
+    assert walsh_transform("x1 XOR x2") == [0, 0, 0, 4]
+
+def test_walsh_or_two_variables():
+    assert walsh_transform("x1 OR x2") == [-2, 2, 2, 2]
+
+
+def test_walsh_xor_three_variables():
+    assert walsh_transform("x1 XOR x2 XOR x3") == [0, 0, 0, 0, 0, 0, 0, 8]