Better typing for DNA base methods

haem.pyi

@@ -61,12 +61,22 @@ class DNABase:
     def __invert__(self) -> DNABase:
         """See `DNABase.complement`."""
 
-    def __add__(self, other: typing.Union[DNABase, DNASequence, str]) -> DNASequence:
+    def __add__(
+        self,
+        other: typing.Union[
+            DNABase, DNASequence, typing.Iterator[str], typing.Sequence[str], str
+        ],
+    ) -> DNASequence:
         """Create a new sequence consisting of this base followed by the given
         sequence member(s)."""
         ...
 
-    def __radd__(self, other: typing.Union[DNABase, DNASequence, str]) -> DNASequence:
+    def __radd__(
+        self,
+        other: typing.Union[
+            DNABase, DNASequence, str, typing.Iterator[str], typing.Sequence[str], str
+        ],
+    ) -> DNASequence:
         """Create a new sequence consisting of the given sequence member(s)
         followed by this base."""
         ...

tests/test_dna_base.py

@@ -179,11 +179,19 @@ def test__bool__(base: haem.DNABase, result: bool) -> None:
         (haem.DNABase("A"), haem.DNASequence(), haem.DNASequence("A")),
         (haem.DNABase("A"), "", haem.DNASequence("A")),
         (haem.DNABase("A"), haem.DNASequence("T"), haem.DNASequence("AT")),
+        (haem.DNABase("A"), iter("ACG"), haem.DNASequence("AACG")),
+        (haem.DNABase("A"), ["A", "C", "G"], haem.DNASequence("AACG")),
     ],
 )
 def test__add__(
     left: haem.DNABase,
-    right: typing.Union[haem.DNABase, haem.DNASequence, str],
+    right: typing.Union[
+        haem.DNABase,
+        haem.DNASequence,
+        typing.Iterator[str],
+        typing.Sequence[str],
+        str,
+    ],
     result: haem.DNASequence,
 ) -> None:
     assert left + right == result
@@ -196,6 +204,8 @@ def test__add__(
         (haem.DNASequence(), haem.DNABase("A"), haem.DNASequence("A")),
         ("", haem.DNABase("A"), haem.DNASequence("A")),
         ("T", haem.DNABase("A"), haem.DNASequence("TA")),
+        (iter("ACG"), haem.DNABase("A"), haem.DNASequence("ACGA")),
+        (["A", "C", "G"], haem.DNABase("A"), haem.DNASequence("ACGA")),
     ],
 )
 def test__radd__(