Fixup test_context.py unittest import