diff --git a/tests/test_stdin.py b/tests/test_stdin.py index 212b309..2ed346e 100644 --- a/tests/test_stdin.py +++ b/tests/test_stdin.py @@ -1,9 +1,17 @@ """Tests for polyfill's stdin monkey patching.""" import flake8 -import pep8 -import pycodestyle import pytest +try: + import pep8 +except ImportError: + pep8 = None + +try: + import pycodestyle +except ImportError: + pycodestyle = None + from flake8_polyfill import stdin from flake8_polyfill import version