gh-115282: Fix direct invocation of test_traceback.py (#115283)

sobolevn · web-flow · commit cc573c70b7d5 · 2024-02-11T19:07:08.000+03:00
diff --git a/Lib/test/test_traceback.py b/Lib/test/test_traceback.py
@@ -3124,10 +3124,13 @@ def test_smoke_user_exception(self):
         class MyException(Exception):
             pass
 
-        self.do_test_smoke(
-            MyException('bad things happened'),
-            ('test.test_traceback.TestTracebackException.'
-             'test_smoke_user_exception.<locals>.MyException'))
+        if __name__ == '__main__':
+            expected = ('TestTracebackException.'
+                        'test_smoke_user_exception.<locals>.MyException')
+        else:
+            expected = ('test.test_traceback.TestTracebackException.'
+                        'test_smoke_user_exception.<locals>.MyException')
+        self.do_test_smoke(MyException('bad things happened'), expected)
 
     def test_from_exception(self):
         # Check all the parameters are accepted.
