Add future imports for with statement use in tests