From 0d93315e34d86549524959ba5fec98f7818779cd Mon Sep 17 00:00:00 2001
From: Ryan Williams <ryan.williams@tiledb.com>
Date: Wed, 6 Mar 2024 21:23:59 -0500
Subject: [PATCH] wip test: verify pa.ChunkedArray slice NDSparseArray

---
 apis/python/tests/test_sparse_nd_array.py | 37 +++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/apis/python/tests/test_sparse_nd_array.py b/apis/python/tests/test_sparse_nd_array.py
index 0c405858a0..cfbf588165 100644
--- a/apis/python/tests/test_sparse_nd_array.py
+++ b/apis/python/tests/test_sparse_nd_array.py
@@ -901,6 +901,43 @@ def test_sparse_nd_array_table_slicing(tmp_path, io, write_format, read_format):
         assert not bad
 
 
+def test_sparse_nd_array_slice_types(tmp_path):
+    io = {
+        "name": "2D coords=(np64[1, 2], np32[3, 4])",
+        "shape": (9, 11),
+        "coords": (
+            pa.chunked_array([[1], [2]]),
+            # np.array([1, 2], dtype=np.uint64),
+            np.array([3, 4], dtype=np.uint32),
+        ),
+        "dims": {
+            "soma_dim_0": [1, 1, 2, 2],
+            "soma_dim_1": [3, 4, 3, 4],
+        },
+        "throws": None,
+    }
+
+    # Set up contents
+    arrow_tensor = create_random_tensor(
+        format="coo",
+        shape=io["shape"],
+        dtype=np.float64(),  # float32 not preserved through IO round-trip
+        density=1.0,
+    )
+
+    with soma.SparseNDArray.create(
+            tmp_path.as_posix(), type=pa.float64(), shape=io["shape"]
+    ) as snda_w:
+        snda_w.write(arrow_tensor)
+
+    with soma.SparseNDArray.open(tmp_path.as_posix()) as snda:
+        table = snda.read(io["coords"]).tables().concat()
+        for column_name in table.column_names:
+            if column_name in io["dims"]:
+                assert table[column_name].to_pylist() == io["dims"][column_name]
+
+
+
 @pytest.mark.parametrize(
     ("result_order", "want"),
     [