Remove filter by filename in utils.filenames_from

1 files changed, 10 insertions, 0 deletions

diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py
index dcbf8b8..76b3e50 100644
--- a/tests/unit/test_utils.py
+++ b/tests/unit/test_utils.py
@@ -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'))