common.py 1.61 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
Theo Steininger's avatar
Theo Steininger committed
13 14 15 16 17
#
# Copyright(C) 2013-2017 Max-Planck-Society
#
# NIFTy is being developed at the Max-Planck-Institut fuer Astrophysik
# and financially supported by the Studienstiftung des deutschen Volkes.
18

Jait Dixit's avatar
Jait Dixit committed
19
from nose_parameterized import parameterized
20 21
from nifty import RGSpace, LMSpace, HPSpace, GLSpace, PowerSpace
from nifty.config import dependency_injector as di
Jait Dixit's avatar
Jait Dixit committed
22 23 24 25 26


def custom_name_func(testcase_func, param_num, param):
    return "%s_%s" % (
        testcase_func.__name__,
27
        parameterized.to_safe_name("_".join(str(x) for x in param.args)),
Jait Dixit's avatar
Jait Dixit committed
28 29
    )

30 31 32 33

def expand(*args, **kwargs):
    return parameterized.expand(*args, testcase_func_name=custom_name_func,
                                **kwargs)
34 35 36


def generate_spaces():
37
    spaces = [RGSpace(4), PowerSpace(RGSpace((4, 4), harmonic=True)),
Theo Steininger's avatar
Theo Steininger committed
38 39 40
              LMSpace(5), HPSpace(4)]
    if 'pyHealpix' in di:
        spaces.append(GLSpace(4))
41 42
    return spaces

43

Theo Steininger's avatar
Theo Steininger committed
44 45
def generate_harmonic_spaces():
    spaces = [RGSpace(4, harmonic=True), LMSpace(5)]
46
    return spaces