add utils to check val in adata.obs

scverse · michalk8 · Apr 14, 2022 · Sep 16, 2021 · Sep 17, 2021 · Sep 21, 2021
commit fed18523c9304715c4819c8ed1488a8477ea31fe
diff --git a/squidpy/pl/_utils.py b/squidpy/pl/_utils.py
@@ -620,3 +620,13 @@ def _dendrogram(data: NDArrayA, method: str, **kwargs: Any) -> tuple[list[int],
 def _sanitize_anndata(adata: AnnData) -> None:
     """Transform string annotations to categoricals."""
     adata._sanitize()
+
+
+def _assert_value_in_obs(adata: AnnData, key: str, val: Sequence[Any] | Any) -> None:
+    if key not in adata.obs:
+        raise KeyError(f"Cluster key `{key}` not found in `adata.obs`.")
+    if not isinstance(val, Sequence):
+        val = [val]
+    val = list(set(val) - set(adata.obs[key].unique()))
+    if len(val) != 0:
+        raise ValueError(f"Values `{val}` not found in `adata.obs[{key}]`.")