update tests

oxli-bio · Sep 24, 2024 · 0d19e68 · 0d19e68
1 parent e5f774d
commit 0d19e68
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/python/tests/test_attr.py b/src/python/tests/test_attr.py
@@ -67,7 +67,7 @@ def test_consumed_after_count():
 
 def test_consumed_after_consume():
     kmer_table = oxli.KmerCountTable(ksize=16)
-    kmer_table.consume("ACGTACGXACGTACGT", allow_bad_kmers=True)  # Length is 16
+    kmer_table.consume("ACGTACGXACGTACGT", skip_bad_kmers=True)  # Length is 16
     assert (
         kmer_table.consumed == 16
     ), "consumed should be updated to 16 after consuming sequence"

diff --git a/src/python/tests/test_basic.py b/src/python/tests/test_basic.py
@@ -77,22 +77,22 @@ def test_consume_bad_DNA():
     cg = oxli.KmerCountTable(4)
     seq = "ATCGGX"
     with pytest.raises(ValueError, match="bad k-mer encountered at position 2"):
-        cg.consume(seq, allow_bad_kmers=False)
+        cg.consume(seq, skip_bad_kmers=False)
 
 
 def test_consume_bad_DNA_2():
     # test an invalid base in first position
     cg = oxli.KmerCountTable(4)
     seq = "XATCGG"
     with pytest.raises(ValueError, match="bad k-mer encountered at position 0"):
-        cg.consume(seq, allow_bad_kmers=False)
+        cg.consume(seq, skip_bad_kmers=False)
 
 
 def test_consume_bad_DNA_ignore():
     # we can ignore bad DNA
     cg = oxli.KmerCountTable(4)
     seq = "XATCGG"
-    print(cg.consume(seq, allow_bad_kmers=True))
+    print(cg.consume(seq, skip_bad_kmers=True))
     assert cg.get("ATCG") == 1
     assert cg.get("TCGG") == 1
     assert cg.get("CCGA") == 1  # rc