test: new traceback test cases

Author: krisztian-gajdar

Lib/test/test_traceback.py

@@ -3803,7 +3803,7 @@ def test_traceback_header(self):
         self.assertEqual(list(exc.format()), ["Exception: haven\n"])
 
     def test_name_error_punctuation_with_suggestions(self):
-        def raise_mssage(message, name):
+        def raise_mssage(message, name, name_from=None):
             try:
                 raise NameError(message, name=name)
             except NameError as e:
@@ -3822,6 +3822,19 @@ def raise_mssage(message, name):
                 messsage = raise_mssage(message, name)
                 self.assertEqual(messsage, expected)
 
+        with self.subTest("combined suggestion"):
+            messsage = raise_mssage("foo", "abc")
+            expected_message = (
+                "NameError: foo. Did you mean: 'abs'? "
+                "Or did you forget to import 'abc'?\n"
+            )
+            self.assertEqual(messsage, expected_message)
+
+        with self.subTest("'did you mean' suggestion"):
+            messsage = raise_mssage("bar", "flaot")
+            expected_message = "NameError: bar. Did you mean: 'float'?\n"
+            self.assertEqual(messsage, expected_message)
+
     def test_import_error_punctuation_handling_with_suggestions(self):
         def raise_mssage(message):
             try: