From 9c409d1d0fc45ebf79e0a398e629c7e334ab55e5 Mon Sep 17 00:00:00 2001
From: Martin Becker <1241516+mgbckr@users.noreply.github.com>
Date: Thu, 25 Jan 2024 09:25:44 +0100
Subject: [PATCH] Fix tests

---
 src/spacec/plotting/__init__.py | 12 ++++++++++--
 tests/__init__.py               |  0
 tests/test_3_clustering.py      |  4 ++--
 3 files changed, 12 insertions(+), 4 deletions(-)
 create mode 100644 tests/__init__.py

diff --git a/src/spacec/plotting/__init__.py b/src/spacec/plotting/__init__.py
index 87230b9..3679039 100644
--- a/src/spacec/plotting/__init__.py
+++ b/src/spacec/plotting/__init__.py
@@ -9,13 +9,19 @@
     stacked_bar_plot,
     zcount_thres,
 )
-from ._qptiff_converter import tissue_lables
-from ._segmentation import segmentation_ch, show_masks
+from ._qptiff_converter import (
+    tissue_lables
+)
+from ._segmentation import (
+    segmentation_ch, 
+    show_masks
+)
 
 __all__ = [
     # segmentation
     "segmentation_ch",
     "show_masks",
+
     # general
     "coordinates_on_image",
     "catplot",
@@ -27,5 +33,7 @@
     "dumbbell",
     "count_patch_proximity_res",
     "zcount_thres",
+
+    # qptiff converter
     "tissue_lables",
 ]
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/test_3_clustering.py b/tests/test_3_clustering.py
index 42ec7b3..23975ae 100644
--- a/tests/test_3_clustering.py
+++ b/tests/test_3_clustering.py
@@ -278,7 +278,7 @@ def test_3_clustering():
         )
 
         # cell type percentage tab and visualization [much few]
-        ct_perc_tab, _ = sc.pl.pl_stacked_bar_plot_ad(
+        ct_perc_tab, _ = sp.pl.stacked_bar_plot(
             adata=adata,  # adata object to use
             color="celltype_fine",  # column containing the categories that are used to fill the bar plot
             grouping="condition",  # column containing a grouping variable (usually a condition or cell group)
@@ -293,7 +293,7 @@ def test_3_clustering():
             norm=False,  # if True, then whatever plotted will be scaled to sum of 1
         )
 
-        sc.pl.pl_create_pie_charts_ad(
+        sp.pl.create_pie_charts(
             adata,
             color="celltype_fine",
             grouping="condition",