Add to_matrix() to assertion

juanjosegarciaripoll · May 7, 2024 · 87e9093 · 87e9093
1 parent 4971e16
commit 87e9093
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_truncate/test_simplify_mpo.py b/tests/test_truncate/test_simplify_mpo.py
@@ -63,6 +63,6 @@ def test_simplification_method(self):
             mpo = MPO([t.reshape(t.shape[0], d, d, t.shape[-1]) for t in ψ._data])
             φ0 = simplify_mpo(mpo, strategy=strategy_0)
             φ1 = simplify_mpo(mpo, strategy=strategy_1)
-            self.assertSimilar(ψ, φ0)
-            self.assertSimilar(ψ, φ1)
-            self.assertSimilar(φ0, φ1)
+            self.assertSimilar(mpo.to_matrix(), φ0.to_matrix())
+            self.assertSimilar(mpo.to_matrix(), φ1.to_matrix())
+            self.assertSimilar(φ0.to_matrix(), φ1.to_matrix())