Add test_plotting.py::TestGPCCA::test_plot_tsi

theislab · Mar 2, 2024 · 67597b2 · 67597b2
1 parent b2c3e1e
commit 67597b2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/tests/_ground_truth_figures/plot_tsi.png b/tests/_ground_truth_figures/plot_tsi.png
diff --git a/tests/test_plotting.py b/tests/test_plotting.py
@@ -2257,6 +2257,13 @@ def test_scvelo_transition_matrix_projection(self, mc: GPCCA, fpath: str):
             save=fpath,
         )
 
+    @compare(kind="gpcca")
+    def test_plot_tsi(self, mc: GPCCA, fpath: str):
+        terminal_states = ["Neuroblast", "Astrocyte", "Granule mature"]
+        cluster_key = "clusters"
+        _ = mc.tsi(n_macrostates=3, terminal_states=terminal_states, cluster_key=cluster_key, n_cells=10)
+        mc.plot_tsi(dpi=DPI, save=fpath)
+
 
 class TestLineage:
     @compare(kind="lineage")