summaryrefslogtreecommitdiffstats
path: root/pym
diff options
context:
space:
mode:
authorZac Medico <zmedico@gentoo.org>2011-01-31 14:28:12 -0800
committerZac Medico <zmedico@gentoo.org>2011-01-31 14:28:12 -0800
commit72aa86e97367849e7dd867d0320bbf0b91c836a6 (patch)
tree1e15ea1ba3cce86eb0be24eb739d0623aa511026 /pym
parentee02b5b7d747d2b5b5d5339c91dee0cd3f2ee150 (diff)
downloadportage-72aa86e97367849e7dd867d0320bbf0b91c836a6.tar.gz
portage-72aa86e97367849e7dd867d0320bbf0b91c836a6.tar.bz2
portage-72aa86e97367849e7dd867d0320bbf0b91c836a6.zip
ImportModulesTestCase: use iterator, not list
Diffstat (limited to 'pym')
-rw-r--r--pym/portage/tests/lint/test_import_modules.py9
1 files changed, 3 insertions, 6 deletions
diff --git a/pym/portage/tests/lint/test_import_modules.py b/pym/portage/tests/lint/test_import_modules.py
index 87bb4bf07..8d257c5a6 100644
--- a/pym/portage/tests/lint/test_import_modules.py
+++ b/pym/portage/tests/lint/test_import_modules.py
@@ -13,7 +13,7 @@ class ImportModulesTestCase(TestCase):
expected_failures = frozenset((
))
- for mod in self._list_modules(PORTAGE_PYM_PATH):
+ for mod in self._iter_modules(PORTAGE_PYM_PATH):
try:
__import__(mod)
except ImportError as e:
@@ -21,8 +21,7 @@ class ImportModulesTestCase(TestCase):
self.assertTrue(False, "failed to import '%s': %s" % (mod, e))
del e
- def _list_modules(self, base_dir):
- all_modules = []
+ def _iter_modules(self, base_dir):
for parent, dirs, files in os.walk(base_dir):
parent = _unicode_decode(parent,
encoding=_encodings['fs'], errors='strict')
@@ -38,6 +37,4 @@ class ImportModulesTestCase(TestCase):
x = parent_mod
else:
x = parent_mod + "." + x
- all_modules.append(x)
-
- return all_modules
+ yield x