Removed invalid TypeWarning import from test_main.py