From 930b5ceffcb3599a1f8d41c8093d1efcf595e87f Mon Sep 17 00:00:00 2001 From: Christian Herdtweck Date: Tue, 9 Oct 2018 09:44:32 +0200 Subject: [PATCH] Fix iter_helper import and test --- src/iter_helpers.py | 11 ++--------- test/test_iter_helpers.py | 4 ++-- 2 files changed, 4 insertions(+), 11 deletions(-) diff --git a/src/iter_helpers.py b/src/iter_helpers.py index 36c2991..9ea3f23 100644 --- 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 --- 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)) -- 1.7.1