Skip to content

Commit 1531824

Browse files
Add more tests.
1 parent c4655e3 commit 1531824

1 file changed

Lines changed: 21 additions & 2 deletions

File tree

Lib/test/test_traceback.py

Lines changed: 21 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4255,15 +4255,18 @@ class A:
42554255
analization = None
42564256
fiⁿₐˡᵢᶻₐᵗᵢᵒₙ = None
42574257

4258-
self.assertIn("'finalization'", self.get_suggestion(A(), 'fiⁿₐˡᵢᶻₐᵗᵢᵒₙ'))
4258+
suggestion = self.get_suggestion(A(), 'fiⁿₐˡᵢᶻₐᵗᵢᵒₙ')
4259+
self.assertIn("'finalization'", suggestion)
4260+
self.assertNotIn("analization", suggestion)
42594261

42604262
class B:
42614263
attr_a = None
4262-
attr_µ = None
4264+
attr_µ = None # attr_\xb5
42634265

42644266
suggestion = self.get_suggestion(B(), 'attr_\xb5')
42654267
self.assertIn("'attr_\u03bc'", suggestion)
42664268
self.assertIn(r"'attr_\u03bc'", suggestion)
4269+
self.assertNotIn("attr_a", suggestion)
42674270

42684271

42694272
class GetattrSuggestionTests(BaseSuggestionTests):
@@ -4898,6 +4901,22 @@ def foo(self):
48984901
instance = A()
48994902
actual = self.get_suggestion(instance.foo)
49004903
self.assertIn("self.finalization", actual)
4904+
self.assertNotIn("fiⁿₐˡᵢᶻₐᵗᵢᵒₙ", actual)
4905+
self.assertNotIn("analization", actual)
4906+
4907+
class B:
4908+
def __init__(self):
4909+
self.attr_µ = None # attr_\xb5
4910+
def foo(self):
4911+
attr_a = 1
4912+
x = attr_µ # attr_\xb5
4913+
4914+
instance = B()
4915+
actual = self.get_suggestion(instance.foo)
4916+
self.assertIn("self.attr_\u03bc", actual)
4917+
self.assertIn(r"self.attr_\u03bc", actual)
4918+
self.assertNotIn("attr_\xb5", actual)
4919+
self.assertNotIn("attr_a", actual)
49014920

49024921
def test_unbound_local_error_with_instance(self):
49034922
class A:

0 commit comments

Comments
 (0)