setup.py

import os
import sys
import io
from setuptools import setup, find_packages
import distutils
from contextlib import redirect_stdout
from importlib import import_module
sys.path.insert(0, os.path.abspath('doc'))
from version_from_git import version_number_from_git


quick_tests = [
    'test_serial_scenarios.test_ldc_mrt',
    'test_serial_scenarios.test_channel_srt',
]


class SimpleTestRunner(distutils.cmd.Command):
    """A custom command to run selected tests"""

    description = 'run some quick tests'
    user_options = []

    @staticmethod
    def _run_tests_in_module(test):
        """Short test runner function - to work also if py.test is not installed."""
        test = 'lbmpy_tests.' + test
        mod, function_name = test.rsplit('.', 1)
        if isinstance(mod, str):
            mod = import_module(mod)

        func = getattr(mod, function_name)
        print("   -> %s in %s" % (function_name, mod.__name__))
        with redirect_stdout(io.StringIO()):
            func()

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        """Run command."""
        for test in quick_tests:
            self._run_tests_in_module(test)


setup(name='lbmpy',
      version=version_number_from_git(),
      description='Code Generation for Lattice Boltzmann Methods',
      author='Martin Bauer',
      license='AGPLv3',
      author_email='martin.bauer@fau.de',
      url='https://i10git.cs.fau.de/pycodegen/lbmpy/',
      packages=['lbmpy'] + ['lbmpy.' + s for s in find_packages('lbmpy')],
      install_requires=['pystencils'],
      classifiers=[
            'Development Status :: 4 - Beta',
            'Framework :: Jupyter',
            'Topic :: Software Development :: Code Generators',
            'Topic :: Scientific/Engineering :: Physics',
            'Intended Audience :: Developers',
            'Intended Audience :: Science/Research',
            'License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)',
      ],
      python_requires=">=3.6",
      extras_require={
            'gpu': ['pystencils[gpu]'],
            'alltrafos': ['pystencils[alltrafos]'],
            'interactive': ['pystencils[interactive]', 'scipy', 'scikit-image', 'cython'],
            'doc': ['pystencils[doc]'],
      },
      cmdclass={
          'quicktest': SimpleTestRunner
      }
      )