update set_sequence_register tests

Author: FilomenoSanchez

conkit/core/contactmap.py

@@ -539,7 +539,7 @@ def set_sequence_register(self, altloc=False):
             else:
                 res1_index = c.res1_seq
                 res2_index = c.res2_seq
-            if res1_index <= seq_len or res2_index <= seq_len:
+            if res1_index <= seq_len and res2_index <= seq_len:
                 c.res1 = self.sequence.seq[res1_index - 1]
                 c.res2 = self.sequence.seq[res2_index - 1]
             else:

conkit/core/tests/test_contactmap.py

@@ -432,6 +432,22 @@ def test_set_sequence_register_2(self):
             [("A", "E"), ("B", "F"), ("A", "D"), ("C", "F"), ("B", "E")], [(c.res1, c.res2) for c in contact_map1]
         )
 
+    def test_set_sequence_register_3(self):
+        contact_map1 = ContactMap("1")
+        for comb in [(1, 5, 1.0), (2, 6, 1.0), (1, 4, 1.0), (3, 6, 1.0), (2, 5, 1.0)]:
+            contact_map1.add(Contact(*comb))
+        contact_map1.sequence = Sequence("foo", "ABCDE")
+        with self.assertRaises(ValueError):
+            contact_map1.set_sequence_register()
+
+    def test_set_sequence_register_4(self):
+        contact_map1 = ContactMap("1")
+        for comb in [(1, 5, 1.0), (6, 3, 1.0), (1, 4, 1.0), (3, 2, 1.0), (2, 5, 1.0)]:
+            contact_map1.add(Contact(*comb))
+        contact_map1.sequence = Sequence("foo", "ABCDE")
+        with self.assertRaises(ValueError):
+            contact_map1.set_sequence_register()
+
     def test_match_1(self):
         contact_map1 = ContactMap("foo")
         for params in [(1, 5, 1.0), (1, 6, 1.0), (2, 7, 1.0), (3, 5, 1.0), (2, 8, 1.0)]: