Fix tests

patrick-kidger · Aug 9, 2024 · c3af158 · c3af158
1 parent ead1e90
commit c3af158
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/helpers.py b/tests/helpers.py
@@ -133,6 +133,8 @@ def params(only_pseudo):
                 continue
             if make_operator is make_diagonal_operator and tags != lx.diagonal_tag:
                 continue
+            if make_operator is make_identity_operator and tags != lx.unit_diagonal_tag:
+                continue
             if (
                 make_operator is make_tridiagonal_operator
                 and tags != lx.tridiagonal_tag