SylvainDe / DidYouMean-Python

                self.runs(set1, after)

    def test_unmatched_msg(self):

        """Test that arbitrary strings are supported."""

        code = 'raise AttributeError("unmatched ATTRIBUTEERROR")'

        self.throws(code, UNKNOWN_ATTRIBUTEERROR)

    # TODO: Add sugg for situation where self/cls is the missing parameter

class TypeErrorTests(GetSuggestionsTests):

    """Class for tests related to TypeError."""

    def test_unhashable(self):

        """Test for UNHASHABLE exception."""

        # NICE_TO_HAVE : suggest hashable equivalent

        self.throws('s = set([list()])', UNHASHABLE)

        self.throws('s = set([dict()])', UNHASHABLE)

        self.throws('s = set([set()])', UNHASHABLE)

        self.runs('s = set([tuple()])')

        self.runs('s = set([frozenset()])')

        self.runs(good_code)

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

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

    def test_nonlocal4(self):

        """suggest close matches to variable name."""

        # NICE_TO_HAVE (needs access to variable in enclosing scope)

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

        code = 'def func():\n\tfoo = 1\n\tdef nested():\n\t\tnonlocal {0}'

        typo, good = 'foob', 'foo'

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

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

        self.runs(good_code, after)

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

        self.throws(bad_code, NOBINDING, [], after)

    def test_nonlocal_at_module_level(self):

        """nonlocal must be used in function."""

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