Have ContigBuilder.add() default to adding 1x the bases provided (#76)

fulcrumgenomics · Nov 23, 2023 · 66b2200 · 66b2200
1 parent b8ac915
commit 66b2200
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/fgpyo/fasta/builder.py b/fgpyo/fasta/builder.py
@@ -96,7 +96,7 @@ def __init__(
         self.species = species
         self.bases = ""
 
-    def add(self, bases: str, times: int) -> "ContigBuilder":
+    def add(self, bases: str, times: int = 1) -> "ContigBuilder":
         """
         Method for adding bases to a new or existing instance of ContigBuilder.
 

diff --git a/fgpyo/fasta/tests/test_builder.py b/fgpyo/fasta/tests/test_builder.py
@@ -17,6 +17,13 @@ def test_overrides_FastaBuilder() -> None:
     assert builder.line_length == 90
 
 
+def test_bases_length_from_ContigBuilder_add_default() -> None:
+    """Checks that the default addition of bases is a single addition."""
+    builder = FastaBuilder()
+    builder.add("chr10").add("AAAA")
+    assert len(builder.__getitem__("chr10").bases) == 4
+
+
 @pytest.mark.parametrize(
     "name, bases, times, length_bases",
     [