test both solutions

KohlbacherLab · Nov 29, 2023 · 4c84a4c · 4c84a4c
1 parent 0adb12e
commit 4c84a4c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/epytope/test/TestSpacerDesign.py b/epytope/test/TestSpacerDesign.py
@@ -33,7 +33,10 @@ def test_standart_functions(self):
         sbws = EpitopeAssemblyWithSpacer(self.epis, cl_pred, epi_pred, self.alleles, solver="glpk")
         sol = sbws.solve()
         print(sol)
-        assert all(i == str(j) for i, j in zip(["GHRMAWDM", "WWQW", "VYEADDVI"], sol))
+        test1 = all(i == str(j) for i, j in zip(["GHRMAWDM", "WWQW", "VYEADDVI"], sol))
+        test2 = all(i == str(j) for i, j in zip(["GHRMAWDM", "WWRW", "VYEADDVI"], sol))
+
+        assert (test1 or test2)
 
     def test_unsupported_allele_length_combination(self):
         """