Added required future imports to test file.