Remove filter by filename in utils.filenames_from

This commit is contained in:
Yevhen Amelin 2019-03-18 20:20:03 +02:00
parent d365e06d7e
commit 6fb49b2e80
2 changed files with 12 additions and 3 deletions

View file

@ -377,9 +377,8 @@ def filenames_from(arg, predicate=None):
for filename in files:
joined = os.path.join(root, filename)
if predicate(joined) or predicate(filename):
continue
yield joined
if not predicate(joined):
yield joined
else:
yield arg

View file

@ -186,6 +186,16 @@ def test_filenames_from_a_directory_with_a_predicate():
assert 'flake8/__init__.py' not in filenames
def test_filenames_from_a_directory_with_a_predicate_from_the_current_dir():
"""Verify that predicates filter filenames_from."""
filenames = list(utils.filenames_from(
arg='./src/flake8',
predicate=lambda filename: filename == '__init__.py',
))
assert len(filenames) > 2
assert './src/flake8/__init__.py' in filenames
def test_filenames_from_a_single_file():
"""Verify that we simply yield that filename."""
filenames = list(utils.filenames_from('flake8/__init__.py'))