Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
  • Loading branch information
kounelisagis committed Jan 28, 2025
1 parent 2fafced commit fb18dd4
Showing 1 changed file with 8 additions and 15 deletions.
23 changes: 8 additions & 15 deletions tiledb/tests/test_libtiledb.py
Original file line number Diff line number Diff line change
Expand Up @@ -2278,22 +2278,16 @@ def test_sparse_mixed_domain_uint_float64(self, fx_sparse_cell_order):
],
)
def test_sparse_string_domain(
self, coords, expected_ned, allows_duplicates, sparse_cell_order
self, coords, expected_ned, allows_duplicates, fx_sparse_cell_order
):
# if sparse_cell_order in ("hilbert", "row-major", "col-major") and allows_duplicates == True:
# if tiledb.libtiledb.version() < (2, 8):
# pytest.xfail(
# "Skipping known bug with legacy reader and empty strings"
# )

path = self.path("sparse_string_domain")
dom = tiledb.Domain(tiledb.Dim(name="d", domain=(None, None), dtype=np.bytes_))
att = tiledb.Attr(name="a", dtype=np.int64)
schema = tiledb.ArraySchema(
domain=dom,
attrs=(att,),
sparse=True,
cell_order=sparse_cell_order,
cell_order=fx_sparse_cell_order,
allows_duplicates=allows_duplicates,
capacity=10000,
)
Expand All @@ -2302,30 +2296,29 @@ def test_sparse_string_domain(
data = [1, 2, 3, 4][: len(coords)]

with tiledb.open(path, "w") as A:
breakpoint()
A[coords] = data

with tiledb.open(path) as A:
ned = A.nonempty_domain()[0]
self.assertEqual(A.nonempty_domain(), ((tuple(expected_ned)),))
assert_array_equal(A.nonempty_domain(), ((tuple(expected_ned)),))

if not (
sparse_cell_order in ("hilbert", "row-major", "col-major")
fx_sparse_cell_order in ("hilbert", "row-major", "col-major")
and allows_duplicates == True
):
assert_array_equal(A[ned[0] : ned[1]]["a"], data)
self.assertEqual(set(A[ned[0] : ned[1]]["d"]), set(coords))

if allows_duplicates and sparse_cell_order != "hilbert":
res_u1 = A.query(order="U").multi_index[ned[0] : ned[1]]
if allows_duplicates and fx_sparse_cell_order != "hilbert":
res_u1 = A.query().multi_index[ned[0] : ned[1]]
assert_array_equal(res_u1["a"], data)
self.assertEqual(set(res_u1["d"]), set(coords))

res_u2 = A.query(order="U")[ned[0] : ned[1]]
res_u2 = A.query()[ned[0] : ned[1]]
assert_array_equal(res_u2["a"], data)
self.assertEqual(set(res_u2["d"]), set(coords))

def test_sparse_string_domain2(self, sparse_cell_order):
def test_sparse_string_domain2(self, fx_sparse_cell_order):
path = self.path("sparse_string_domain2")
with self.assertRaises(ValueError):
dims = [
Expand Down

0 comments on commit fb18dd4

Please sign in to comment.