update test note and 5mer naming to 4mer

iqbal-lab-org · Jan 27, 2021 · cad5729 · cad5729
1 parent 545a2e0
commit cad5729
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/from_msa/test_cluster_sequences.py b/tests/from_msa/test_cluster_sequences.py
@@ -383,10 +383,10 @@ def test_GivenAllSequencesSmallEditDist_ReturnsNoClustering(self):
             self.assertEqual(expected_clustering, result)
 
     def test_GivenManyVeryDifferentSequences_EachSeqInOwnCluster(self):
-        # all 256 distinct DNA 5-mers.
-        # We want each in one cluster, but do not want to run clustering 255 times
-        all_5mers = list(map("".join, product(standard_bases, repeat=4)))
-        alignment = make_alignment(all_5mers)
+        # all 256 distinct DNA 4-mers.
+        # We want clustering to keep looking for clusters, and stop at MAX_CLUSTERS
+        all_4mers = list(map("".join, product(standard_bases, repeat=4)))
+        alignment = make_alignment(all_4mers)
         result = kmeans_cluster_seqs_in_interval([0, 4], alignment, 4)
         self.assertEqual(len(result), MAX_CLUSTERS)