diff --git a/tests/tests.py b/tests/tests.py index 7a017e3..c81134b 100644 --- a/tests/tests.py +++ b/tests/tests.py @@ -7,13 +7,17 @@ """ import sys -import collections if sys.version_info[0:2] == (2, 6): import unittest2 as unittest else: import unittest +if sys.version_info[0] == 3 and sys.version_info[1] > 2: + from collections.abc import Iterable, Iterator +else: + from collections import Iterable, Iterator + from iso639 import Iso639, languages from examples.logic import map_language from pycountry import languages as pclanguages @@ -112,8 +116,8 @@ def test_len(self): self.assertEqual(len(languages), 7981) def test_iter(self): - self.assertIsInstance(languages, collections.Iterable) - self.assertIsInstance(iter(languages), collections.Iterator) + self.assertIsInstance(languages, Iterable) + self.assertIsInstance(iter(languages), Iterator) def test_exceptions(self): self.assertRaises(KeyError, languages.get, part1='En') # Wrong value