fix test & iterator

polyply/src/gen_dna.py

@@ -30,7 +30,6 @@ def _dna_edge_iterator(meta_molecule, source):
     while True:
         neighbors = meta_molecule.neighbors(source)
         src_resid = meta_molecule.nodes[source]["resid"]
-        count += 1
         for next_node in neighbors:
             next_resid = meta_molecule.nodes[next_node]["resid"]
             diff = src_resid - next_resid
@@ -42,9 +41,8 @@ def _dna_edge_iterator(meta_molecule, source):
             if next_resid > src_resid and next_node == first_node:
                 yield (source, next_node)
                 return
-
-            if count > len(meta_molecule.nodes):
-                return
+        else:
+            return
 
 def complement_dsDNA(meta_molecule):
     """

polyply/tests/test_gen_dna.py

@@ -61,11 +61,10 @@ def test_complement_dsDNA(extra_edges, termini, expect_ter):
             assert new_edge_attrs[(idx, jdx)] == edge_attr[(idx, jdx)]
         else:
             assert new_edge_attrs[(idx, jdx)] == edge_attr[(jdx, idx)]
-
     assert len(list(nx.connected_components(test_DNA_mol))) == 2
 
 
-def test_complement_dsDNA():
+def test_complement_dsDNA_error():
     """
     Test that an unkown residue generates an error when
     passed to gen_dna processor.
@@ -87,6 +86,4 @@ def test_complement_dsDNA():
            "at the moment only allows base-pair completion for molecules that only "
            "consist of dsDNA. Please conact the developers if you whish to create a "
            "more complicated molecule.")
-
     assert str(context.value) == msg
-