Update test_plot_cpdb.py

zktuong · Jul 2, 2024 · e74207e · e74207e
1 parent b3f9a50
commit e74207e
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/tests/test_plot_cpdb.py b/tests/test_plot_cpdb.py
@@ -326,6 +326,30 @@ def test_plot_exclude(mock_show, adata, means, pvals, exclude):
     g
 
 
+@patch("matplotlib.pyplot.show")
+@pytest.mark.usefixtures("adata", "means", "pvals")
+@pytest.mark.parametrize(
+    "int_pair",
+    [
+        "CXCL13-CXCR5",
+        ["CXCL13-CXCR5"],
+        "CXCL13_CXCR5",
+        ["CXCL13_CXCR5"],
+    ],
+)
+def test_plot_interacting_pairs(mock_show, adata, means, pvals, int_pair):
+    g = plot_cpdb(
+        adata=adata,
+        cell_type1="B cell",
+        cell_type2="CD4T cell",
+        means=means,
+        pvals=pvals,
+        celltype_key="celltype",
+        interacting_pairs=int_pair,
+    )
+    g
+
+
 @patch("matplotlib.pyplot.show")
 @pytest.mark.usefixtures("adata_v5", "means_v5", "pvals_v5")
 def test_plot_cpdb_v5_interaction_min(mock_show, adata_v5, means_v5, pvals_v5, interaction_scores_v5):