Update test_plot_cpdb_heatmap.py

zktuong · Feb 14, 2024 · 57fad2d · 57fad2d
1 parent 124df69
commit 57fad2d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_plot_cpdb_heatmap.py b/tests/test_plot_cpdb_heatmap.py
@@ -71,3 +71,13 @@ def test_plot_cpdb_heatmap_return(adata, pvals):
     for d in dfs:
         assert isinstance(dfs[d], pd.DataFrame)
         assert dfs[d].shape[0] > 0
+
+
+@patch("matplotlib.pyplot.show")
+@pytest.mark.usefixtures("adata", "pvals")
+def test_plot_cpdb_heatmap_celltypes(mock_show, adata, pvals):
+    g = plot_cpdb_heatmap(
+        pvals=pvals,
+        cell_types=["CD4 T cell", "CD8 T cell", "B cell"],
+    )
+    g