Added Greek letters to valid einsum symbols, and a test that ensures …

…all einsum symbols get parsed properly
pydata · Jan 25, 2025 · ee4ac11 · ee4ac11
1 parent 2bca00c
commit ee4ac11
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/sparse/numba_backend/_common.py b/sparse/numba_backend/_common.py
@@ -19,7 +19,7 @@
     normalize_axis,
 )
 
-_EINSUM_SYMBOLS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+_EINSUM_SYMBOLS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZαβγδϵζηθικλμνξπρστυϕχψωΓΔΕΘΛΠΡΣΦΨΩ"
 _EINSUM_SYMBOLS_SET = set(_EINSUM_SYMBOLS)
 
 

diff --git a/sparse/numba_backend/tests/test_einsum.py b/sparse/numba_backend/tests/test_einsum.py
@@ -82,6 +82,16 @@
 ]
 
 
+@pytest.mark.parametrize("symbols", sparse.numba_backend._common._EINSUM_SYMBOLS_SET)
+def test_symbols(symbols):
+
+    arr = sparse.random(shape=10, density=1.0)
+
+    # ensure we can use any of the defined symbols
+    for symbol in symbols:
+        sparse.einsum(f"{symbol}->{symbol}", arr)
+
+
 @pytest.mark.parametrize("subscripts", einsum_cases)
 @pytest.mark.parametrize("density", [0.1, 1.0])
 def test_einsum(subscripts, density):