test(operation): add tests for operation module

Author: gausshj

test/test_parity_tensor.py

@@ -0,0 +1,20 @@
+from parity_tensor import ParityTensor
+
+import torch
+
+def test_operation() -> None:
+    x = ParityTensor((False, False), ((2, 2), (1, 3)), torch.randn([4, 4])).update_mask()
+
+    print()
+    print(x)
+    print()
+    print(x.permute((0, 1)))
+    print()
+    print(x.permute((1, 0)))
+    print()
+    print(x + 1)
+    print()
+    print(x * 2)
+    print()
+    print(x / 2)
+