Cleaned up wagtailembeds/tests.py imports