From 57fad2d896db57c6100a4e89cb2b48d313d96a73 Mon Sep 17 00:00:00 2001
From: Zewen Kelvin Tuong <z.tuong@uq.edu.au>
Date: Wed, 14 Feb 2024 10:41:42 +1000
Subject: [PATCH] Update test_plot_cpdb_heatmap.py

---
 tests/test_plot_cpdb_heatmap.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/tests/test_plot_cpdb_heatmap.py b/tests/test_plot_cpdb_heatmap.py
index 0270bd5..6dfb4af 100644
--- 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