Iter

CHANGELOG.md

@@ -6,6 +6,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 
 ## [Unreleased]
 ### Added
+- `__iter__` overload
 - `double_runs` property
 - `repeats` method
 - `name` property

opr/primer.py

@@ -103,6 +103,14 @@ def __str__(self):
         """
         return self._sequence
 
+    def __iter__(self):
+        """
+        Iterate through Primer.
+
+        :return: base as Generator[str]
+        """
+        yield from self.sequence
+
     def reverse(self, inplace=False):
         """
         Reverse sequence.

tests/test_cache.py

@@ -43,4 +43,5 @@ def test_double_runs():
     double_runs_2nd = {}
     for pair in pairs:
         double_runs_2nd[pair] = p1.double_runs[pair]
-    assert len(double_runs_2nd) == len(double_runs) and all(double_runs[pair] == double_runs_2nd[pair] for pair in double_runs)
+    assert len(double_runs_2nd) == len(double_runs) and all(
+        double_runs[pair] == double_runs_2nd[pair] for pair in double_runs)

tests/test_calculations.py

@@ -72,7 +72,19 @@ def test_double_runs():
     p1 = Primer("ATATCGAACACACACACA")
     double_runs = p1.double_runs
     print(double_runs)
-    true_answer = {'GT': 0, 'CA': 5, 'AT': 2, 'TA': 0, 'GC': 0, 'GA': 0, 'AG': 0, 'TG': 0, 'CG': 0, 'TC': 0, 'AC': 5, 'CT': 0}
+    true_answer = {
+        'GT': 0,
+        'CA': 5,
+        'AT': 2,
+        'TA': 0,
+        'GC': 0,
+        'GA': 0,
+        'AG': 0,
+        'TG': 0,
+        'CG': 0,
+        'TC': 0,
+        'AC': 5,
+        'CT': 0}
     assert len(true_answer) == len(double_runs) and all(double_runs[pair] == true_answer[pair] for pair in double_runs)
 
 

tests/test_operations.py

@@ -74,3 +74,11 @@ def test_str():
     oprimer_2 = Primer("ATCGC")
     oprimer_concat = oprimer_1 + oprimer_2
     assert str(oprimer_1) + str(oprimer_2) == oprimer_concat.sequence
+
+
+def test_iter():
+    oprimer_1 = Primer("ATCG")
+    sequence = ""
+    for base in oprimer_1:
+        sequence += base
+    assert oprimer_1.sequence == sequence

tests/test_property.py

@@ -5,7 +5,7 @@
 
 def test_sequence():
     oprimer = Primer("ATCGATCGATCGATCGAT")
-    assert oprimer.sequence== "ATCGATCGATCGATCGAT"
+    assert oprimer.sequence == "ATCGATCGATCGATCGAT"
 
 
 def test_name():