import six assert not six.PY2 assert six.PY3