CountVector: Add __iter__ convenience wrapper

coala · May 4, 2015 · 3e1f1c3 · 3e1f1c3 · sils · May 4, 2015
1 parent 067bd49
commit 3e1f1c3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/bears/codeclone_detection/CountVector.py b/bears/codeclone_detection/CountVector.py
@@ -38,3 +38,6 @@ def __repr__(self):
 
     def __len__(self):
         return len(self.count_vector)
+
+    def __iter__(self):
+        return iter(self.count_vector)
diff --git a/bears/tests/codeclone_detection/CountVectorTest.py b/bears/tests/codeclone_detection/CountVectorTest.py
@@ -42,13 +42,14 @@ def test_weighting(self):
         uut.count_reference(True, False)
         self.assertEqual(uut.count_vector, [2, 2])
 
-    def test_conversion(self):
+    def test_conversions(self):
         uut = CountVector("varname",
                           [lambda cursor, stack: cursor and stack],
                           [2])
         uut.count_reference(True, True)
         self.assertEqual(repr(uut), str(uut))
         self.assertEqual(repr(uut), "[2]")
+        self.assertEqual(list(uut), [2])
 
 
 if __name__ == '__main__':