setup.py

__author__ = 'junz'

from setuptools import setup, find_packages
from setuptools.command.test import test as TestCommand
import io
import os
import sys

here = os.path.abspath(os.path.dirname(__file__))

def read(*filenames, **kwargs):
    encoding = kwargs.get('encoding', 'utf-8')
    sep = kwargs.get('sep', '\n')
    buf = []
    for filename in filenames:
        with io.open(filename, encoding=encoding) as f:
            buf.append(f.read())
    return sep.join(buf)

long_description = read('README.md')

def prepend_find_packages(*roots):
    '''
    Recursively traverse nested packages under the root directories
    '''
    packages = []

    for root in roots:
        packages += [root]
        packages += [root + '.' + s for s in find_packages(root)]

    return packages

class PyTest(TestCommand):
    def finalize_options(self):
        TestCommand.finalize_options(self)
        self.test_args = ['--junitxml=result.xml']
        self.test_args_cov = self.test_args + ['--cov=corticalmapping', '--cov-report=term', '--cov-report=html']
        self.test_suite = True

    def run_tests(self):
        import pytest

        try:
            errcode = pytest.main(self.test_args_cov)
        except:
            errcode = pytest.main(self.test_args)
        sys.exit(errcode)

setup(
    name='corticalmapping',
    version = '2.0.0',
    url='http://stash.corp.alleninstitute.org/users/junz/repos/corticalmapping/',
    author='Jun Zhuang',
    tests_require=['pytest'],
    install_requires=['numpy','scipy', 'PyDAQmx','scikit-image','tifffile'],
    cmdclass={'test': PyTest},
    author_email='junz@alleninstitute.org',
    description='cortical mapping tools',
    long_description=long_description,
    packages=prepend_find_packages('corticalmapping'),
    include_package_data=True,
    package_data={'':['*.md', '*.txt', '*.cfg', '*.hdf5']},
    platforms='any',
    classifiers = [
        'Programming Language :: Python',
        'Development Status :: 4 - Beta',
        'Natural Language :: English',
        'Operating System :: OS Independent',
        ],
    extras_require={
        'testing': ['pytest'],
    }
)