[python] Domain-at-create unit-test PR 2/N (#3190)

single-cell-data · Oct 18, 2024 · fc5b7bb · fc5b7bb
1 parent 7336955
commit fc5b7bb
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/apis/python/tests/test_type_system.py b/apis/python/tests/test_type_system.py
@@ -135,16 +135,21 @@ def test_bool_arrays(tmp_path, bool_array):
         ]
     )
     index_column_names = ["soma_joinid"]
+
+    n_data = len(bool_array)
+    data = {
+        "soma_joinid": list(range(n_data)),
+        "b": bool_array,
+    }
+    rb = pa.Table.from_pydict(data)
+    nrb = len(rb)
+
     with soma.DataFrame.create(
-        tmp_path.as_posix(), schema=schema, index_column_names=index_column_names
+        tmp_path.as_posix(),
+        schema=schema,
+        index_column_names=index_column_names,
+        domain=[[0, max(nrb, 1) - 1]],
     ) as sdf:
-        n_data = len(bool_array)
-
-        data = {
-            "soma_joinid": list(range(n_data)),
-            "b": bool_array,
-        }
-        rb = pa.Table.from_pydict(data)
         sdf.write(rb)
 
     with soma.DataFrame.open(tmp_path.as_posix()) as sdf:

diff --git a/apis/python/tests/test_unicode.py b/apis/python/tests/test_unicode.py
@@ -46,6 +46,7 @@ def sample_dataframe_path(tmp_path, sample_arrow_table):
         tmp_path.as_posix(),
         schema=sample_arrow_table.schema,
         index_column_names=["soma_joinid"],
+        domain=((0, len(sample_arrow_table) - 1),),
     ) as sdf:
         sdf.write(sample_arrow_table)
     return sdf.uri