[python] Experiment-level shaping [WIP] [skip ci]

apis/python/src/tiledbsoma/_sparse_nd_array.py

@@ -313,12 +313,18 @@ def resize(
             self._handle.resize(newshape)
             return (True, "")
 
-    def tiledbsoma_upgrade_shape(self, newshape: Sequence[Union[int, None]]) -> None:
+    def tiledbsoma_upgrade_shape(
+        self, newshape: Sequence[Union[int, None]], check_only: bool = False
+    ) -> StatusAndReason:
         """Allows the array to have a resizeable shape as described in the TileDB-SOMA
         1.15 release notes.  Raises an error if the new shape exceeds maxshape in
         any dimension. Raises an error if the array already has a shape.
         """
-        self._handle.tiledbsoma_upgrade_shape(newshape)
+        if check_only:
+            return self._handle.tiledbsoma_can_upgrade_shape(newshape)
+        else:
+            self._handle.tiledbsoma_upgrade_shape(newshape)
+            return (True, "")
 
     def write(
         self,

apis/python/src/tiledbsoma/_tdb_handles.py

@@ -469,6 +469,12 @@ def tiledbsoma_upgrade_shape(self, newshape: Sequence[Union[int, None]]) -> None
         """Not implemented for DataFrame."""
         raise NotImplementedError
 
+    def tiledbsoma_can_upgrade_shape(
+        self, newshape: Sequence[Union[int, None]]
+    ) -> StatusAndReason:
+        """Not implemented for DataFrame."""
+        raise NotImplementedError
+
     def resize_soma_joinid_shape(self, newshape: int) -> None:
         """Only implemented for DataFrame."""
         raise NotImplementedError
@@ -672,6 +678,17 @@ def tiledbsoma_upgrade_shape(self, newshape: Sequence[Union[int, None]]) -> None
         """
         self._handle.tiledbsoma_upgrade_shape(newshape)
 
+    def tiledbsoma_can_upgrade_shape(
+        self, newshape: Sequence[Union[int, None]]
+    ) -> StatusAndReason:
+        """Allows the array to have a resizeable shape as described in the TileDB-SOMA
+        1.15 release notes.  Raises an error if the new shape exceeds maxshape in
+        any dimension. Raises an error if the array already has a shape.
+        """
+        return cast(
+            StatusAndReason, self._handle.tiledbsoma_can_upgrade_shape(newshape)
+        )
+
 
 class _DictMod(enum.Enum):
     """State machine to keep track of modifications to a dictionary.

apis/python/src/tiledbsoma/io/__init__.py

@@ -20,6 +20,11 @@
     to_anndata,
     to_h5ad,
 )
+from .shaping import (
+    resize_experiment,
+    show_experiment_shapes,
+    upgrade_experiment_shapes,
+)
 
 __all__ = (
     "add_matrix_to_collection",
@@ -37,5 +42,8 @@
     "update_matrix",
     "update_obs",
     "update_var",
+    "upgrade_experiment_shapes",
+    "show_experiment_shapes",
+    "resize_experiment",
     "ExperimentAmbientLabelMapping",
 )