modified plotting

scimap/plotting/heatmap.py

@@ -317,6 +317,7 @@ def plot_category_heatmap_vectorized(data,
                 os.makedirs(saveDir)
             full_path = os.path.join(saveDir, fileName)
             plt.savefig(full_path, dpi=300)
+            plt.close(fig)
             print(f"Saved heatmap to {full_path}")
         else:
             plt.show()

scimap/tests/test_pl.py

@@ -9,7 +9,8 @@
 import pytest
 import os
 import anndata as ad
-import matplotlib.pyplot as plt
+import unittest
+
 
 # load data
 @pytest.fixture
@@ -22,7 +23,13 @@ def adata():
 # heatmap
 def test_heatmap (adata):
     from scimap.plotting.heatmap import heatmap
-    heatmap(adata, groupBy='phenotype', standardScale='column')
+    saveDir = os.getcwd() + '/testFigures'
+    fileName = 'heatmap.png'
+    heatmap(adata, groupBy='phenotype', standardScale='column', saveDir=saveDir, fileName=fileName)
+    # check the file exist
+    full_path = os.path.join(saveDir, fileName)
+    assert os.path.exists(full_path), f"File was not created: {full_path}"
+