Suite et fixup, sets for multiple structural

alignementseq_tests.py

@@ -125,7 +125,7 @@ class BalibaseTestCase(unittest.TestCase):
         structs = []
 
         for i, record in enumerate(seqs):
-            if i >= head:
+            if head and i >= head:
                 break
 
             for extension, fformat, parse in self.formats:
@@ -173,18 +173,6 @@ class BalibaseTestCase(unittest.TestCase):
                 continue
             yield seqs[0], seqs[1], structs[0], structs[1], seq_fname
 
-
-    def get_dataset_records(self):
-        """Generator function to iterate over the record generator  of each
-        unaligned fasta file.
-        """
-        dataset_dir = os.path.join(self.balibase_path, "RV11.unaligned")
-        for filename in os.listdir(dataset_dir):
-            records = Bio.SeqIO.parse(
-                    os.path.join(dataset_dir, filename),
-                    "fasta")
-            yield records, filename
-
 class AlignmentSeqTestCase(BalibaseTestCase):
     """ TestCase for alignment correctness and quality
     """
@@ -372,10 +360,10 @@ class AlignmentSeqTestCase(BalibaseTestCase):
         for in_filename in os.listdir(dataset_dir):
 
             # Read input
-            sequences = list(Bio.SeqIO.parse(
-                os.path.join(dataset_dir, in_filename),
-                "fasta"
-                ))
+            sequences, structs = self.get_seqs_and_structs(
+                os.path.join(dataset_dir, in_filename))
+            if structs is None:
+                continue
 
             # Create output for all but clustalo
             for method, name in zip(methods_test, methods_test_names):