def style_guide_mock(): """Create a mock StyleGuide object.""" return mock.MagicMock( **{ "options.diff": False, "options.jobs": JobsArgument("4"), } )
import os import pep8 import sys try: from StringIO import StringIO except ImportError: from io import StringIO try: from flake8.main.options import JobsArgument jobs = JobsArgument('1') except ImportError: jobs = '1' from flake8.api.legacy import get_style_guide # Quck hack again, 3d time flake8 would be removed, if no volounters found from django.conf import settings from . import set_option class Reporter(object): """ Runs flake8 on python files. """ def add_arguments(self, parser): parser.add_argument('--max-complexity', dest='flake8-max-complexity', type=int, help='McCabe complexity treshold') parser.add_argument( "--pep8-exclude",
def test_jobs_argument_str(): """Test that JobsArgument has a correct __str__.""" assert str(JobsArgument("auto")) == "auto" assert str(JobsArgument("123")) == "123"
def test_parse_valid_jobs_argument(s, is_auto, n_jobs): """Test that --jobs properly parses valid arguments.""" jobs_opt = JobsArgument(s) assert is_auto == jobs_opt.is_auto assert n_jobs == jobs_opt.n_jobs