Import StringIO from io when using Python 3