setup.py

#!/usr/bin/env python

import os
import sys
from setuptools import setup


def setup_package():
    src_path = os.path.dirname(os.path.abspath(sys.argv[0]))
    old_path = os.getcwd()
    os.chdir(src_path)
    sys.path.insert(0, src_path)

    try:
        os.environ['__in-setup'] = '1'  # ensures only version is imported
        from upsetplot import __version__ as version
        # See also setup.cfg
        setup(name='UpSetPlot',
              version=version,
              packages=["upsetplot"],
              setup_requires=['pytest-runner'],
              tests_require=['pytest>=2.7', 'pytest-cov<2.6'],
              # TODO: check versions
              install_requires=['pandas', 'matplotlib>=2.0'])
    finally:
        del sys.path[0]
        os.chdir(old_path)
    return


if __name__ == '__main__':
    setup_package()