TST: Update tests and test images

robertopreste · Mar 4, 2020 · b13dcc9 · b13dcc9
1 parent e8244e6
commit b13dcc9
Show file tree

Hide file tree

Showing 8 changed files with 23 additions and 2 deletions.
diff --git a/mitoviz/tests/images/multisample_linear_split_1.png b/mitoviz/tests/images/multisample_linear_split_1.png
diff --git a/mitoviz/tests/images/multisample_linear_split_2.png b/mitoviz/tests/images/multisample_linear_split_2.png
diff --git a/mitoviz/tests/images/multisample_linear_split_3.png b/mitoviz/tests/images/multisample_linear_split_3.png
diff --git a/mitoviz/tests/images/multisample_linear_split_4.png b/mitoviz/tests/images/multisample_linear_split_4.png
diff --git a/mitoviz/tests/images/multisample_linear_split_5.png b/mitoviz/tests/images/multisample_linear_split_5.png
diff --git a/mitoviz/tests/images/sample_linear_hf_split.png b/mitoviz/tests/images/sample_linear_hf_split.png
diff --git a/mitoviz/tests/images/sample_linear_split.png b/mitoviz/tests/images/sample_linear_split.png
diff --git a/mitoviz/tests/test_variant.py b/mitoviz/tests/test_variant.py
@@ -64,6 +64,9 @@ def test__is_insertion_true(self):
         self.assertTrue(self.variant_ins._is_insertion)
         self.assertTrue(self.variant_ins_raw._is_insertion)
 
+    def test_color(self):
+        self.assertEqual("#2e8b57", self.variant.color)
+
     def test_label(self):
         self.assertEqual("3308C>A", self.variant.label)
         self.assertEqual("3308C>A", self.variant_raw.label)
@@ -88,5 +91,23 @@ def test_linear_x(self):
     def test_linear_y(self):
         self.assertEqual(0.3, self.variant.linear_y)
 
-    def test_color(self):
-        self.assertEqual("#2e8b57", self.variant.color)
+    def test_strand(self):
+        # Given/When
+        variant_h = self.variant
+        variant_l = _Variant(
+            reference="C",
+            position=500,
+            alternate=Substitution("SNV", "A"),
+            hf=0.3
+        )
+        variant_nc = _Variant(
+            reference="C",
+            position=3305,
+            alternate=Substitution("SNV", "A"),
+            hf=0.3
+        )
+
+        # Then
+        self.assertEqual("H", variant_h.strand)
+        self.assertEqual("L", variant_l.strand)
+        self.assertEqual("", variant_nc.strand)