SylvainDe / DidYouMean-Python

        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_keyword_arg_multiples_instances(self):

        """If multiple functions are found, suggestions should be unique."""

        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'

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

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

        self.runs(good_code)

    def test_typo_in_method2(self):

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

        code = 'from math import e, {0}, log'

        typo, sugg = 'pie', 'pi'

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

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