Further max sequence length improvements

Author: fonors

fastonex.py

@@ -67,7 +67,7 @@ def nexus_matrix(seqdict):
     Takes a dictionary containing sequences and returns the NEXUS MATRIX block.
     """
     nexus_matrix_block = "MATRIX\n\n"
-    maxseqlen = max(len(seq) for seq in seqdict)
+    maxseqlen = max(len(seqdict[seq]) for seq in seqdict)
 
     for seq in seqdict:
         if len(seqdict[seq]) < maxseqlen:

test_fastonex.py

@@ -47,7 +47,7 @@ def test_nexus_matrix():
     """
     Unit test that determines if a Leave NEXUS MATRIX block can be determined from a sequence dictionary.
     """
-    known_input = {"Seq1" : "atgc", "Seq2" : "atg-"}
-    expected_output = "MATRIX\n\nSeq1     atgc\nSeq2     atg-\n;\n\nEND;"
+    known_input = {"Seq1" : "atgcatg", "Seq2" : "atg----"}
+    expected_output = "MATRIX\n\nSeq1     atgcatg\nSeq2     atg----\n;\n\nEND;"
 
     assert fastonex.nexus_matrix(known_input) == expected_output