updated tests to sample centre of plots for consensus seq

KatyBrown · May 22, 2024 · 537e14d · 537e14d
1 parent f505217
commit 537e14d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/consensusSeqTest.py b/tests/consensusSeqTest.py
@@ -186,9 +186,9 @@ def tearDown(self):
                            ['tests/test_files/example2.fasta', 'aa']])
     def testPlotResidueFrequencies(self, fasta, typ):
         alignment, names = readMSA(fasta)
-        expected = image.imread("tests/test_files/expected_resfreq_%s.png" % typ)
+        expected = image.imread("tests/test_files/expected_resfreq_%s.png" % typ)[500:1000, 500:1000]
         consensusSeq.plotResidueFrequencies(alignment, typ, self.dest)
-        imi = image.imread(self.dest)
+        imi = image.imread(self.dest)[500:1000, 500:1000]
         simi = skimage.metrics.structural_similarity(expected,
                                                      imi,
                                                      channel_axis=-1,
@@ -207,9 +207,9 @@ def tearDown(self):
     @parameterized.expand([['tests/test_files/example1.fasta']])
     def testResidueChangeCount(self, fasta):
         alignment, names = readMSA(fasta)
-        expected = image.imread("tests/test_files/expected_changecount.png")
+        expected = image.imread("tests/test_files/expected_changecount.png")[500:1000, 500:1000]
         consensusSeq.residueChangeCount(alignment, 'nt', self.dest)
-        imi = image.imread(self.dest)
+        imi = image.imread(self.dest)[500:1000, 500:1000]
         simi = skimage.metrics.structural_similarity(expected,
                                                      imi,
                                                      channel_axis=-1,