1 files changed, 67 insertions, 0 deletions
diff --git a/tests/functional/c/consider_using_enumerate.py b/tests/functional/c/consider_using_enumerate.py
new file mode 100644
index 000000000..758375c37
--- /dev/null
+++ b/tests/functional/c/consider_using_enumerate.py
@@ -0,0 +1,67 @@
+"""Emit a message for iteration through range and len is encountered."""
+
+# pylint: disable=missing-docstring, import-error, useless-object-inheritance, unsubscriptable-object, too-few-public-methods
+
+def bad():
+    iterable = [1, 2, 3]
+    for obj in range(len(iterable)): # [consider-using-enumerate]
+        yield iterable[obj]
+    for obj in range(0, len(iterable)): # [consider-using-enumerate]
+        yield iterable[obj]
+
+
+class Bad(object):
+
+    def __iter__(self):
+        iterable = [1, 2, 3]
+        for i in range(len(iterable)): # [consider-using-enumerate]
+            yield iterable[i]
+
+    def test(self):
+        for i in range(len(self)): # [consider-using-enumerate]
+            yield self[i]
+
+
+def good():
+    iterable = other_obj = [1, 2, 3]
+    total = 0
+    for obj in range(len(iterable)):
+        total += obj
+        yield total
+        yield iterable[obj + 1: 2]
+        yield iterable[len(obj)]
+    for obj in iterable:
+        yield iterable[obj - 1]
+
+    for index, obj in enumerate(iterable):
+        yield iterable[index]
+    for index in range(0, 10):
+        yield iterable[index + 1]
+    for index in range(10):
+        yield iterable[index]
+    for index in range(len([1, 2, 3, 4])):
+        yield index
+    for index in range(1, len(iterable)):
+        yield index
+    for index in range(len(iterable)):
+        yield [1, 2, 3][index]
+        yield len([1, 2, 3])
+    for index in range(len(iterable)):
+        yield other_obj[index]
+
+    from unknown import unknown
+    for index in range(unknown(iterable)):
+        yield iterable[index]
+
+    for index in range(len(iterable)):
+        def test(iterable):
+            return iterable[index]
+        yield test([1, 2, 3])
+
+
+class Good(object):
+
+    def __iter__(self):
+        # Should not suggest enumerate on self
+        for i in range(len(self)):
+            yield self[i]