implement tests for a list of gene symbols/identifiers & string conve…

…rsions
monarch-initiative · May 23, 2024 · 9944a96 · 9944a96
1 parent a195702
commit 9944a96
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/tests/test_analysis.py b/tests/test_analysis.py
@@ -422,8 +422,8 @@ def setUp(self) -> None:
                     rank=2,
                 ),
                 RankedPhEvalGeneResult(
-                    gene_symbol="PLXNA1",
-                    gene_identifier="ENSG00000114554",
+                    gene_symbol="['PLXNA1', 'GENE2']",
+                    gene_identifier="['ENSG00000114554', 'ENSG00000100000']",
                     score=0.8764,
                     rank=4,
                 ),
@@ -772,6 +772,17 @@ def test_assess_gene_prioritisation_threshold_meets_cutoff(self):
             ),
         )
 
+    def test__check_string_representation_string(self):
+        self.assertEqual(
+            self.assess_gene_prioritisation._check_string_representation("GENE1"), "GENE1"
+        )
+
+    def test__check_string_representation_list(self):
+        self.assertEqual(
+            self.assess_gene_prioritisation._check_string_representation("['GENE1', 'GENE2']"),
+            ["GENE1", "GENE2"],
+        )
+
 
 class TestAssessVariantPrioritisation(unittest.TestCase):
     def setUp(self) -> None: