Fix unused imports in test_schema.py.