tests/test_registration_mappings.py WIP [skip ci]

single-cell-data · Oct 9, 2024 · f269ee9 · f269ee9
1 parent c819f13
commit f269ee9
Showing 1 changed file with 40 additions and 10 deletions.
diff --git a/apis/python/tests/test_registration_mappings.py b/apis/python/tests/test_registration_mappings.py
@@ -383,16 +383,6 @@ def test_multiples_without_experiment(
             var_field_name=var_field_name,
         )
 
-    print()
-    print()
-    print()
-    print()
-    print()
-    print(rd)
-    print()
-    print()
-    print()
-    print()
     assert rd.obs_axis.id_mapping_from_values(["AGAG", "GGAG"]).data == (2, 8)
     assert rd.var_axes["measname"].id_mapping_from_values(["ESR1", "VEGFA"]).data == (
         2,
@@ -449,6 +439,15 @@ def test_multiples_without_experiment(
         h5ad_file_names[permutation[2]],
         h5ad_file_names[permutation[3]],
     ]:
+        if tiledbsoma._flags.NEW_SHAPE_FEATURE_FLAG_ENABLED:
+            if tiledbsoma.Experiment.exists(experiment_uri):
+                tiledbsoma.io.resize_experiment(
+                    experiment_uri,
+                    nobs=rd.get_obs_shape(),
+                    nvars=rd.get_var_shapes(),
+                )
+
+        # XXX FIXME
         tiledbsoma.io.from_h5ad(
             experiment_uri,
             h5ad_file_name,
@@ -705,6 +704,13 @@ def test_append_items_with_experiment(obs_field_name, var_field_name):
 
     original = adata2.copy()
 
+    if tiledbsoma._flags.NEW_SHAPE_FEATURE_FLAG_ENABLED:
+        tiledbsoma.io.resize_experiment(
+            soma1,
+            nobs=rd.get_obs_shape(),
+            nvars=rd.get_var_shapes(),
+        )
+
     with tiledbsoma.Experiment.open(soma1, "w") as exp1:
         tiledbsoma.io.append_obs(
             exp1,
@@ -828,6 +834,14 @@ def test_append_with_disjoint_measurements(
         var_field_name=var_field_name,
     )
 
+    if tiledbsoma._flags.NEW_SHAPE_FEATURE_FLAG_ENABLED:
+        # XXX FIXME
+        tiledbsoma.io.resize_experiment(
+            soma_uri,
+            nobs=rd.get_obs_shape(),
+            nvars=rd.get_var_shapes(),
+        )
+
     tiledbsoma.io.from_anndata(
         soma_uri,
         anndata2,
@@ -1176,6 +1190,15 @@ def test_enum_bit_width_append(tmp_path, all_at_once, nobs_a, nobs_b):
         tiledbsoma.io.from_anndata(
             soma_uri, adata, measurement_name=measurement_name, registration_mapping=rd
         )
+
+        if tiledbsoma._flags.NEW_SHAPE_FEATURE_FLAG_ENABLED:
+            # XXX FIXME
+            tiledbsoma.io.resize_experiment(
+                soma_uri,
+                nobs=rd.get_obs_shape(),
+                nvars=rd.get_var_shapes(),
+            )
+
         tiledbsoma.io.from_anndata(
             soma_uri, bdata, measurement_name=measurement_name, registration_mapping=rd
         )
@@ -1191,6 +1214,13 @@ def test_enum_bit_width_append(tmp_path, all_at_once, nobs_a, nobs_b):
             var_field_name=var_field_name,
         )
 
+        if tiledbsoma._flags.NEW_SHAPE_FEATURE_FLAG_ENABLED:
+            tiledbsoma.io.resize_experiment(
+                soma_uri,
+                nobs=rd.get_obs_shape(),
+                nvars=rd.get_var_shapes(),
+            )
+
         tiledbsoma.io.from_anndata(
             soma_uri, bdata, measurement_name=measurement_name, registration_mapping=rd
         )