test orthonormal bases

mehta-lab · Dec 21, 2024 · ba45686 · ba45686
1 parent 55ff94e
commit ba45686
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 20 deletions.
diff --git a/tests/test_util.py b/tests/test_util.py
@@ -44,3 +44,21 @@ def test_pad_zyx_large_padding(zyx_data):
     assert result.shape == (13, 4, 5)
     assert torch.all(result[:5] == 0)
     assert torch.all(result[-5:] == 0)
+
+
+def test_pauli_orthonormal():
+    s = util.pauli()
+    assert torch.allclose(
+        torch.abs(torch.einsum("kij,lji->kl", s, s)) - torch.eye(4),
+        torch.zeros((4, 4)),
+        atol=1e-5,
+    )
+
+
+def test_gellmann_orthonormal():
+    Y = util.gellmann()
+    assert torch.allclose(
+        torch.abs(torch.einsum("kij,lji->kl", Y, Y)) - torch.eye(9),
+        torch.zeros((9, 9)),
+        atol=1e-5,
+    )
diff --git a/waveorder/util.py b/waveorder/util.py
@@ -2258,17 +2258,6 @@ def pauli():
             [[a, 0], [0, a]],
             [[-a, 0], [0, a]],
             [[0, a], [a, 0]],
-            [[0, 1j * a], [-1j * a, 0]],
-        ]
-    )
-    return sigma
-
-
-# torch.allclose(
-#     torch.abs(torch.einsum("kij,lji->kl", s, s) - torch.eye(4)),
-#     torch.zeros((4, 4)),
-#     atol=1e-5,
-# )
 
 
 def gellmann():
@@ -2297,12 +2286,4 @@ def gellmann():
             [[0, 0, c], [0, 0, 0], [c, 0, 0]],
             [[0, 0, 0], [0, -c, 0], [0, 0, c]],  #
         ], dtype=torch.complex64
-    )
-
-    # torch.allclose(
-
-
-#     torch.abs(torch.einsum("kij,lji->kl", Y, Y) - torch.eye(9)),
-#     torch.zeros((9, 9)),
-#     atol=1e-5,
-# )
+    )