Removed unneccessary imports from test_documents.py