SylvainDe / DidYouMean-Python

            code2 = code + end

            for op in ('-', '+'):

                typo, sugg = 2 * op, op + '=1'

                bad_code, good_code = format_str(code + end, typo, sugg)

                self.throws(bad_code, INVALIDSYNTAX)

                self.runs(good_code)

    def test_wrong_bool_operator(self):

        """Trying to use '&&' or '||'."""

        code = 'True {0} False'

        for typo, sugg in (('&&', 'and'), ('||', 'or')):

            bad_code, good_code = format_str(code, typo, sugg)

            self.throws(bad_code, INVALIDSYNTAX, "'" + sugg + "'")

            self.runs(good_code)

    def test_import_future_not_first(self):

        """Test what happens when import from __future__ is not first."""

        code = 'a = 8/7\nfrom __future__ import division'

        self.throws(code, FUTUREFIRST)

    def test_import_future_not_def(self):

        """Should be 'division'."""

        code = 'from __future__ import {0}'

        typo, sugg = 'divisio', 'division'

        bad_code, good_code = format_str(code, typo, sugg)

        self.throws(bad_code, FUTFEATNOTDEF, "'" + sugg + "'")

        # NICE_TO_HAVE

        version = (3, 0)

        code = 'FoobarClass().some_method.{0}'

        attributes = [('im_func', '__func__', []),

                      ('im_self', '__self__', []),

                      ('im_class', '__self__.__class__', ["'__class__'"])]

        for (old_att, new_att, sugg) in attributes:

            old_code, new_code = format_str(code, old_att, new_att)

            self.runs(old_code, up_to_version(version))

            self.throws(old_code, ATTRIBUTEERROR, sugg, from_version(version))

            self.runs(new_code)

    def test_moved_between_str_string(self):

        """Some methods have been moved from string to str."""

        # NICE_TO_HAVE

        version1 = (3, 0)

        version2 = (3, 5)

        code = 'import string\n{0}.maketrans'

        code_str, code_string = format_str(code, 'str', 'string')

        code_str2 = 'str.maketrans'  # No 'string' import

        code_str3 = 'import string as my_string\nstr.maketrans'  # Named import

        self.throws(code_str, ATTRIBUTEERROR, [], up_to_version(version1))

        self.throws(code_str2, ATTRIBUTEERROR, [], up_to_version(version1))

        self.throws(code_str3, ATTRIBUTEERROR, [], up_to_version(version1))

        self.runs(code_string, up_to_version(version1))

        self.throws(code_string, ATTRIBUTEERROR, [], (version1, version2))