Fixed regex matching
diff --git a/pre_commit_hooks/tests_should_end_in_test.py b/pre_commit_hooks/tests_should_end_in_test.py
index 5df6fd8..4bfc767 100644
--- a/pre_commit_hooks/tests_should_end_in_test.py
+++ b/pre_commit_hooks/tests_should_end_in_test.py
@@ -3,6 +3,7 @@
import argparse
import re
import sys
+from os.path import basename
def validate_files(argv=None):
@@ -15,14 +16,13 @@
args = parser.parse_args(argv)
retcode = 0
- test_name_pattern = '.*_test.py'
- if args.django:
- test_name_pattern = 'test.*.py'
+ test_name_pattern = 'test_.*.py' if args.django else '.*_test.py'
for filename in args.filenames:
+ base = basename(filename)
if (
- not re.match(test_name_pattern, filename) and
- not filename.endswith('__init__.py') and
- not filename.endswith('/conftest.py')
+ not re.match(test_name_pattern, base) and
+ not base == '__init__.py' and
+ not base == 'conftest.py'
):
retcode = 1
print(
