SylvainDe / DidYouMean-Python

    def test_iter_cannot_be_interpreted_as_int(self):

        """Trying to call `range(len(iterable))` (bad) and forget the len."""

        before, after = before_and_after((3, 0))

        bad_code = 'range([0, 1, 2])'

        good_code = 'range(len([0, 1, 2]))'

        sugg = "'len(list)'"

        self.runs(good_code)

        self.throws(bad_code, INTEXPECTED, sugg, before)

        self.throws(bad_code, CANNOTBEINTERPRETED, sugg, after)

    RANGE_CODE_TEMPLATES = [

        'range({0})',

        'range({0}, 14)',

        'range(0, 24, {0})'

    ]

    INDEX_CODE_TEMPLATES = ['[1, 2, 3][{0}]', '(1, 2, 3)[{0}]']

    def test_str_cannot_be_interpreted_as_int(self):

        """Forget to convert str to int."""

        before, after = before_and_after((3, 0))

        suggs = ["'int(str)'", "'len(str)'"]

        for code in self.RANGE_CODE_TEMPLATES:

    def test_keyword_builtin(self):

        """A few builtins (like int()) have a different error message."""

        # NICE_TO_HAVE

        # 'max', 'input', 'len', 'abs', 'all', etc have a specific error

        # message and are not relevant here

        before, after = before_and_after((3, 7))

        for builtin, kwarg in [

                ('float', False), ('bool', False),

                ('int', True), ('complex', True)]:

            code = builtin + '(this_doesnt_exist=2)'

            old_exc = UNEXPECTEDKWARG2

            new_exc = UNEXPECTEDKWARG4 if kwarg else NOKWARGS

            sugg = [] if kwarg else NO_KEYWORD_ARG_MSG

            self.throws(code, old_exc, [], before, interpreters='cpython')

            self.throws(code, new_exc, sugg, after, interpreters='cpython')

            self.throws(code, UNEXPECTEDKWARG, interpreters='pypy')

    def test_keyword_builtin_print(self):

        """Builtin "print" has a different error message."""

        # It would be NICE_TO_HAVE suggestions on keyword arguments

        before, mid, after = before_mid_and_after((3, 0), (3, 7))

        self.throws(bad, INVALIDTOKEN, [], after, 'cpython')

        self.throws(bad, INVALIDSYNTAX, [], after, 'pypy')

    def test_extended_unpacking(self):

        """Extended iterable unpacking is added with Python 3."""

        before, after = before_and_after((3, 0))

        code = '(a, *rest, b) = range(5)'

        self.throws(code, INVALIDSYNTAX, [], before)

        self.runs(code, after)

    def test_ellipsis(self):

        """Triple dot (...) aka Ellipsis can be used anywhere in Python 3."""

        before, after = before_and_after((3, 0))

        code = '...'

        self.throws(code, INVALIDSYNTAX, [], before)

        self.runs(code, after)

    def test_fstring(self):

        """Fstring (see PEP 498) appeared in Python 3.6."""