Fix iter_helper import and test

diff --git a/src/iter_helpers.py b/src/iter_helpers.py
index 36c2991..9ea3f23 100644 (file)
--- a/src/iter_helpers.py
+++ b/src/iter_helpers.py
@@ -25,17 +25,10 @@ been deemed useful
 """
 
 import itertools as it
-from .type_helpers import PY2, PY3
 
 
-zip_longest = None
-izip_longest = None
-if PY2:
-    zip_longest = it.izip_longest
-    izip_longest = it.izip_longest
-else:
-    zip_longest = it.zip_longest
-    izip_longest = it.zip_longest
+zip_longest = it.zip_longest
+izip_longest = it.zip_longest
 
 
 def pairwise(iterable):

diff --git a/test/test_iter_helpers.py b/test/test_iter_helpers.py
index 73bd8b8..34e8266 100644 (file)
--- a/test/test_iter_helpers.py
+++ b/test/test_iter_helpers.py
@@ -36,7 +36,7 @@ import unittest
 from src import iter_helpers
 
 
-class ModuleNameTester(unittest.TestCase):
+class IterHelperTester(unittest.TestCase):
     """ only test case in this module, see module doc for more help """
 
     def test_interesting_repetition(self):
@@ -52,7 +52,7 @@ class ModuleNameTester(unittest.TestCase):
         self.assertFalse(iter_helpers.interesting_repetition(8))
         self.assertFalse(iter_helpers.interesting_repetition(9))
         self.assertTrue (iter_helpers.interesting_repetition(10))
-        self.assertTrue (iter_helpers.interesting_repetition(11))
+        self.assertFalse(iter_helpers.interesting_repetition(11))
         self.assertTrue (iter_helpers.interesting_repetition(150))
         self.assertFalse(iter_helpers.interesting_repetition(151))
         self.assertTrue (iter_helpers.interesting_repetition(2000))