Merge pull request #429 from rrauenza/rrauenza-patch-1
fix regular expression for test files
diff --git a/pre_commit_hooks/tests_should_end_in_test.py b/pre_commit_hooks/tests_should_end_in_test.py
index 7a1e7c0..d93595f 100644
--- a/pre_commit_hooks/tests_should_end_in_test.py
+++ b/pre_commit_hooks/tests_should_end_in_test.py
@@ -18,7 +18,7 @@
args = parser.parse_args(argv)
retcode = 0
- test_name_pattern = 'test.*.py' if args.django else '.*_test.py'
+ test_name_pattern = r'test.*\.py' if args.django else r'.*_test\.py'
for filename in args.filenames:
base = os.path.basename(filename)
if (
diff --git a/tests/tests_should_end_in_test_test.py b/tests/tests_should_end_in_test_test.py
index 2acfa17..4df2963 100644
--- a/tests/tests_should_end_in_test_test.py
+++ b/tests/tests_should_end_in_test_test.py
@@ -11,6 +11,10 @@
assert ret == 1
+def test_regex():
+ assert main(('foo_test_py',)) == 1
+
+
def test_main_django_all_pass():
ret = main((
'--django', 'tests.py', 'test_foo.py', 'test_bar.py',