add diag check

jcmgray · Oct 22, 2024 · 78dec01 · 78dec01
1 parent c506c68
commit 78dec01
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_autoray.py b/tests/test_autoray.py
@@ -76,6 +76,9 @@ def gen_rand(shape, backend, dtype="float64"):
         ("cosh", (), ()),
         ("count_nonzero", (), ()),
         ("cumsum", (0,), ("sparse",)),
+        ("diag", (), ("sparse",)),
+        ("diag", (1,), ("sparse",)),
+        ("diag", (-1,), ("sparse",)),
         ("exp", (), ()),
         ("imag", (), ()),
         ("log", (), ()),