Add simple tests for row_annotations.

Author: newmana

stlearn/add.py

@@ -1,12 +1,12 @@
 from .adds.add_deconvolution import add_deconvolution
-from .adds.image import image
-from .adds.labels import labels
 from .adds.add_loupe_clusters import add_loupe_clusters
-from .adds.lr import lr
 from .adds.add_mask import add_mask, apply_mask
-from .adds.positions import positions
 from .adds.annotation import annotation
+from .adds.image import image
+from .adds.labels import labels
+from .adds.lr import lr
 from .adds.parsing import parsing
+from .adds.positions import positions
 
 __all__ = [
     "image",

tests/adds/test_row_annotations.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+
+"""Tests for ClusterPlot."""
+
+import unittest
+
+from stlearn.adds import row_annotations
+from tests.utils import read_test_data, test_data_path
+
+
+class TestRowAnnotations(unittest.TestCase):
+    """Tests for row annotations."""
+
+    @classmethod
+    def setUpClass(cls):
+        cls._base_adata = read_test_data()
+        cls.annotations_path = (
+            f"{test_data_path()}/" + "v1_human_breast_cancer_block_a_section_1.csv"
+        )
+
+
+def setUp(self):
+    """Set up test data with known clusters."""
+    self.adata = self.__class__._base_adata.copy()
+
+
+def test_add_row_annotations(self):
+    row_annotations.row_annotations(self.adata, self.__class__.annotations_path, "ID")
+
+    assert "annot_type" in self.adata.obs.columns
+    assert "fine_annot_type" in self.adata.obs.columns
+
+    # Check annotated the same number
+    annotated = self.adata.obs["annot_type"].dropna()
+    fine_annotated = self.adata.obs["fine_annot_type"].dropna()
+    assert len(fine_annotated) == len(annotated)
+
+
+def test_add_row_annotations_with_missing_column(self):
+    with self.assertRaises(ValueError):
+        row_annotations.row_annotations(
+            self.adata,
+            self.__class__.annotations_path,
+            join_column="nonexistent_column",
+        )

tests/pl/test_cluster_plot.py

@@ -10,9 +10,9 @@
 import pandas as pd
 
 from stlearn.pl.classes import ClusterPlot
-
 from tests.utils import read_test_data
 
+
 class TestClusterPlot(unittest.TestCase):
     """Tests for ClusterPlot."""
 

tests/spatial/test_psts.py

@@ -7,9 +7,9 @@
 import numpy as np
 
 import stlearn as st
-
 from tests.utils import read_test_data
 
+
 class TestPSTS(unittest.TestCase):
     """Tests for `stlearn` package."""
 

tests/spatial/test_sme.py

@@ -9,9 +9,9 @@
 import scanpy as sc
 
 import stlearn as st
-
 from tests.utils import read_test_data
 
+
 class TestSME(unittest.TestCase):
     """Tests for `stlearn` package."""