SylvainDe / DidYouMean-Python

        code = 'class MyClass:\n\tdef func(self, ' + sugg + '):' \

               '\n\t\tpass\nMyClass().func({0}=1)'

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

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

        self.runs(good_code)

    def test_keyword_arg_class_method(self):

        """Should be the same as previous test but on a class method."""

        code = 'class MyClass:\n\t@classmethod\n\tdef func(cls, a):' \

               '\n\t\tpass\nMyClass.func({0}=1)'

        bad_code, good_code = format_str(code, 'babar', 'a')

        self.throws(bad_code, UNEXPECTEDKWARG)

        self.runs(good_code)

    def test_keyword_arg_class_method2(self):

        """Should be the same as previous test but on a class method."""

        typo, sugg = 'abcdf', 'abcdef'

        code = 'class MyClass:\n\t@classmethod ' \

               '\n\tdef func(cls, ' + sugg + '):\n ' \

               '\t\tpass\nMyClass.func({0}=1)'

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

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

        self.runs(good_code)

    def test_no_name_no_sugg(self):

        """No suggestion."""

        self.throws('from math import fsfsdfdjlkf', CANNOTIMPORT)

    def test_wrong_import(self):

        """Should be 'math'."""

        code = 'from {0} import pi'

        typo, sugg = 'itertools', 'math'

        self.assertTrue(sugg in STAND_MODULES)

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

        self.throws(bad_code, CANNOTIMPORT, "'" + good_code + "'")

        self.runs(good_code)

    def test_typo_in_method(self):

        """Should be 'pi'."""

        code = 'from math import {0}'

        typo, sugg = 'pie', 'pi'