From db24e812e7c0a8d57c1c60b5c5deaf71f523d62e Mon Sep 17 00:00:00 2001 From: stefanpeidli Date: Mon, 30 Sep 2024 17:23:26 +0200 Subject: [PATCH] Improve code coverage by label_transfer tests Checks if error is correctly raised when no KNN is present in adata. --- .../_perturbation_space/test_simple_perturbation_space.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/tools/_perturbation_space/test_simple_perturbation_space.py b/tests/tools/_perturbation_space/test_simple_perturbation_space.py index d897bb2b..4475f711 100644 --- a/tests/tools/_perturbation_space/test_simple_perturbation_space.py +++ b/tests/tools/_perturbation_space/test_simple_perturbation_space.py @@ -240,9 +240,12 @@ def test_label_transfer(): adata = AnnData(X) perturbations = np.array(["A", "B", "C"] * 22 + ["unknown"] * 3) adata.obs["perturbation"] = perturbations - sc.pp.neighbors(adata, use_rep="X") - sc.tl.umap(adata) + with pytest.raises(ValueError): + ps = pt.tl.PseudobulkSpace() + ps.label_transfer(adata) + + sc.pp.neighbors(adata, use_rep="X") ps = pt.tl.PseudobulkSpace() ps.label_transfer(adata) assert "unknown" not in adata.obs["perturbation"]