From 78dec013bef224df72d3cd276a08da3a0541a95a Mon Sep 17 00:00:00 2001
From: Johnnie Gray <johnniemcgray@gmail.com>
Date: Mon, 21 Oct 2024 17:02:52 -0700
Subject: [PATCH] add diag check

---
 tests/test_autoray.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/test_autoray.py b/tests/test_autoray.py
index 1e25101..9e02d05 100644
--- 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", (), ()),