Allow using either `unittest2` or `unittest`.