setup.py 2.84 KB
Newer Older
1
2
import os
import sys
Martin Bauer's avatar
Martin Bauer committed
3
import io
4
from setuptools import setup, find_packages
Martin Bauer's avatar
Martin Bauer committed
5
6
7
8
import distutils
from contextlib import redirect_stdout
from importlib import import_module

9
sys.path.insert(0, os.path.abspath('doc'))
Martin Bauer's avatar
Martin Bauer committed
10
11

quick_tests = [
12
    # 'test_serial_scenarios.test_ldc_mrt',
Martin Bauer's avatar
Martin Bauer committed
13
    'test_serial_scenarios.test_channel_srt',
Martin Bauer's avatar
Martin Bauer committed
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
]


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)
        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)
45

Frederik Hennig's avatar
Frederik Hennig committed
46

Dominik Thoennes's avatar
Dominik Thoennes committed
47
48
49
50
51
52
53
54
55
try:
    sys.path.insert(0, os.path.abspath('doc'))
    from version_from_git import version_number_from_git

    version = version_number_from_git()
    with open("RELEASE-VERSION", "w") as f:
        f.write(version)
except ImportError:
    version = open('RELEASE-VERSION', 'r').read()
56

Frederik Hennig's avatar
Frederik Hennig committed
57

58
59
60
61
62
def readme():
    with open('README.md') as f:
        return f.read()


63
setup(name='lbmpy',
Dominik Thoennes's avatar
Dominik Thoennes committed
64
      version=version,
65
      description='Code Generation for Lattice Boltzmann Methods',
66
      long_description=readme(),
67
      long_description_content_type="text/markdown",
68
69
70
      author='Martin Bauer',
      license='AGPLv3',
      author_email='martin.bauer@fau.de',
71
      url='https://i10git.cs.fau.de/pycodegen/lbmpy/',
72
      packages=['lbmpy'] + ['lbmpy.' + s for s in find_packages('lbmpy')],
73
      install_requires=['pystencils', 'sympy>=1.2', 'numpy>=1.11.0'],
74
      classifiers=[
75
76
77
78
79
80
81
          '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+)',
82
83
84
      ],
      python_requires=">=3.6",
      extras_require={
85
86
87
88
89
90
91
          'gpu': ['pycuda'],
          'opencl': ['pyopencl'],
          'alltrafos': ['islpy', 'py-cpuinfo'],
          'interactive': ['scipy', 'scikit-image', 'cython', 'matplotlib',
                          'ipy_table', 'imageio', 'jupyter', 'pyevtk'],
          'doc': ['sphinx', 'sphinx_rtd_theme', 'nbsphinx',
                  'sphinxcontrib-bibtex', 'sphinx_autodoc_typehints', 'pandoc'],
92
      },
Martin Bauer's avatar
Martin Bauer committed
93
94
95
      cmdclass={
          'quicktest': SimpleTestRunner
      }
96
      )