fix tests

unipept · Aug 27, 2024 · b6ff10d · b6ff10d
1 parent 031a609
commit b6ff10d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/sa-index/src/bounds_cache.rs b/sa-index/src/bounds_cache.rs
@@ -112,6 +112,13 @@ mod tests {
 
         for i in 0..20_usize.pow(5) {
             let kmer = kmer_cache.index_to_kmer(i);
+
+            if kmer.contains(&b'L') {
+                let equated_kmer = kmer.iter().map(|&c| if c == b'L' { b'I' } else { c }).collect::<Vec<u8>>();
+                assert_eq!(kmer_cache.kmer_to_index(&kmer), kmer_cache.kmer_to_index(&equated_kmer));
+                continue;
+            }
+
             assert_eq!(kmer_cache.kmer_to_index(&kmer), i);
         }
     }