Added isometry tests to TestCase

juanjosegarciaripoll · Mar 24, 2024 · 401cb8a · 401cb8a
1 parent 29d85ee
commit 401cb8a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/tools.py b/tests/tools.py
@@ -69,6 +69,10 @@ def random_uniform_mps(
             random_uniform_mps(d, size, truncate=truncate, rng=self.rng), **kwdargs
         )
 
+    def assertApproximateIsometry(self, A, direction, places=7) -> None:
+        if not approximateIsometry(A, direction, places):
+            raise self.failureException(f"Tensor is not isometry:\nA={A}")
+
 
 def similar(A, B, **kwdargs):
     if sp.issparse(A):