pocketpy/tests/802_traceback.py

import traceback
import sys

if sys.argv[0].endswith('.pyc'):
    exit()

try:
    a = {'123': 4}
    b = a[6]
except KeyError:
    actual = traceback.format_exc()

expected = '''Traceback (most recent call last):
  File "tests/802_traceback.py", line 9
    b = a[6]
KeyError: 6'''

if actual != expected:
    print('--- ACTUAL RESULT -----')
    print(actual)
    print('--- EXPECTED RESULT ---')
    print(expected)
    exit(1)