Add basic unit test, pass 2

api/python/cellxgene_census/tests/test_get_anndata.py

@@ -158,7 +158,6 @@ def test_get_anndata_wrong_layer_names(census: soma.Collection) -> None:
             assert raise_info.value.args[0] == "Unknown X layer name"
 
 
-@pytest.mark.skip(reason="Enable when obsm is available in a live Census distribution.")
 @pytest.mark.live_corpus
 @pytest.mark.parametrize("obsm_layer", ["scvi", "geneformer"])
 def test_get_anndata_obsm_one_layer(census: soma.Collection, obsm_layer: str) -> None:
@@ -177,7 +176,6 @@ def test_get_anndata_obsm_one_layer(census: soma.Collection, obsm_layer: str) ->
     assert ad.obsm[obsm_layer].shape[0] == 100
 
 
-@pytest.mark.skip(reason="Enable when obsm is available in a live Census distribution.")
 @pytest.mark.live_corpus
 @pytest.mark.parametrize("obsm_layers", [["scvi", "geneformer"]])
 def test_get_anndata_obsm_two_layers(census: soma.Collection, obsm_layers: List[str]) -> None:
@@ -195,3 +193,21 @@ def test_get_anndata_obsm_two_layers(census: soma.Collection, obsm_layers: List[
     for obsm_layer in obsm_layers:
         assert obsm_layer in ad.obsm.keys()
         assert ad.obsm[obsm_layer].shape[0] == 100
+
+@pytest.mark.live_corpus
+@pytest.mark.parametrize("add_obs_embeddings", [["scvi", "geneformer"]])
+def test_get_anndata_add_obs_embeddings(census: soma.Collection, add_obs_embeddings: List[str]) -> None:
+    with census:
+        ad = cellxgene_census.get_anndata(
+            census,
+            organism="Homo sapiens",
+            X_name="raw",
+            obs_coords=slice(100),
+            var_coords=slice(200),
+            add_obs_embeddings=add_obs_embeddings,
+        )
+
+    assert len(ad.obsm.keys()) == 2
+    for obsm_layer in add_obs_embeddings:
+        assert obsm_layer in ad.obsm.keys()
+        assert ad.obsm[obsm_layer].shape[0] == 100