[Finch] API: More creation functions (#669)

Author: mtsokol

ci/array-api-skips.txt

@@ -9,25 +9,24 @@ array_api_tests/test_signatures.py::test_func_signature[unique_counts]
 array_api_tests/test_signatures.py::test_func_signature[unique_inverse]
 array_api_tests/test_signatures.py::test_func_signature[unique_values]
 
-# asarray - missing 'device' keyword
-array_api_tests/test_signatures.py::test_func_signature[asarray]
-
 # creation functions
 array_api_tests/test_signatures.py::test_func_signature[empty]
 array_api_tests/test_signatures.py::test_func_signature[empty_like]
-array_api_tests/test_signatures.py::test_func_signature[full]
-array_api_tests/test_signatures.py::test_func_signature[full_like]
-array_api_tests/test_signatures.py::test_func_signature[ones]
-array_api_tests/test_signatures.py::test_func_signature[ones_like]
-array_api_tests/test_signatures.py::test_func_signature[zeros]
-array_api_tests/test_signatures.py::test_func_signature[zeros_like]
 array_api_tests/test_signatures.py::test_func_signature[arange]
 array_api_tests/test_signatures.py::test_func_signature[linspace]
 array_api_tests/test_signatures.py::test_func_signature[meshgrid]
 array_api_tests/test_signatures.py::test_func_signature[tril]
 array_api_tests/test_signatures.py::test_func_signature[triu]
-# eye - missing 'device' keyword
+
+# missing 'device' keyword
+array_api_tests/test_signatures.py::test_func_signature[asarray]
 array_api_tests/test_signatures.py::test_func_signature[eye]
+array_api_tests/test_signatures.py::test_func_signature[full]
+array_api_tests/test_signatures.py::test_func_signature[full_like]
+array_api_tests/test_signatures.py::test_func_signature[ones]
+array_api_tests/test_signatures.py::test_func_signature[ones_like]
+array_api_tests/test_signatures.py::test_func_signature[zeros]
+array_api_tests/test_signatures.py::test_func_signature[zeros_like]
 
 # inspection functions
 array_api_tests/test_signatures.py::test_func_signature[can_cast]
@@ -83,7 +82,6 @@ array_api_tests/test_signatures.py::test_func_signature[remainder]
 array_api_tests/test_signatures.py::test_func_signature[argmax]
 array_api_tests/test_signatures.py::test_func_signature[argmin]
 array_api_tests/test_signatures.py::test_func_signature[from_dlpack]
-array_api_tests/test_signatures.py::test_array_method_signature[__array_namespace__]
 
 # linalg namespace
 array_api_tests/test_signatures.py::test_extension_func_signature[linalg.cross]
@@ -98,3 +96,16 @@ array_api_tests/test_signatures.py::test_extension_func_signature[linalg.svdvals
 array_api_tests/test_signatures.py::test_extension_func_signature[linalg.tensordot]
 array_api_tests/test_signatures.py::test_extension_func_signature[linalg.vecdot]
 array_api_tests/test_signatures.py::test_extension_func_signature[linalg.vector_norm]
+
+# Array object namespace
+array_api_tests/test_signatures.py::test_array_method_signature[__bool__]
+array_api_tests/test_signatures.py::test_array_method_signature[__complex__]
+array_api_tests/test_signatures.py::test_array_method_signature[__dlpack__]
+array_api_tests/test_signatures.py::test_array_method_signature[__dlpack_device__]
+array_api_tests/test_signatures.py::test_array_method_signature[__float__]
+array_api_tests/test_signatures.py::test_array_method_signature[__index__]
+array_api_tests/test_signatures.py::test_array_method_signature[__int__]
+array_api_tests/test_signatures.py::test_array_method_signature[__setitem__]
+
+# missing 'stream' from signature
+array_api_tests/test_signatures.py::test_array_method_signature[to_device]

pyproject.toml

@@ -38,7 +38,7 @@ tests = [
 ]
 tox = ["sparse[tests]", "tox"]
 all = ["sparse[docs,tox]", "matrepr"]
-finch = ["finch-tensor>=0.1.12"]
+finch = ["finch-tensor>=0.1.14"]
 
 [project.urls]
 Documentation = "https://sparse.pydata.org/"

sparse/finch_backend/__init__.py

@@ -18,6 +18,12 @@
     atan,
     atan2,
     atanh,
+    bitwise_and,
+    bitwise_invert,
+    bitwise_left_shift,
+    bitwise_or,
+    bitwise_right_shift,
+    bitwise_xor,
     bool,
     ceil,
     compiled,
@@ -35,6 +41,8 @@
     float64,
     floor,
     floor_divide,
+    full,
+    full_like,
     int8,
     int16,
     int32,
@@ -51,6 +59,8 @@
     multiply,
     negative,
     nonzero,
+    ones,
+    ones_like,
     permute_dims,
     positive,
     pow,
@@ -72,6 +82,8 @@
     uint32,
     uint64,
     where,
+    zeros,
+    zeros_like,
 )
 
 __all__ = [
@@ -87,6 +99,12 @@
     "atan",
     "atan2",
     "atanh",
+    "bitwise_and",
+    "bitwise_invert",
+    "bitwise_left_shift",
+    "bitwise_or",
+    "bitwise_right_shift",
+    "bitwise_xor",
     "bool",
     "compiled",
     "complex64",
@@ -143,4 +161,10 @@
     "ceil",
     "nonzero",
     "where",
+    "full",
+    "full_like",
+    "ones",
+    "ones_like",
+    "zeros",
+    "zeros_like",
 ]