diff --git a/tests/test_style.py b/tests/test_style.py index b45fb04..87ce734 100644 --- a/tests/test_style.py +++ b/tests/test_style.py @@ -1,8 +1,11 @@ import pep8 +from glob import glob def test_pep8_comformance(): ''' Test that the code conforms to PEP8 ''' pep8style = pep8.StyleGuide(config_file='.pep8') - result = pep8style.check_files(['tsstats.py', 'tests/test_config.py', 'tests/test_general.py', 'tests/test_style.py']) + files = glob('tests/*.py') + files.append('tsstats.py') + result = pep8style.check_files(files) assert result.total_errors == 0