1
2
3
4
5
6
7
8
9
10
11
12
13
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
from functools import wraps
import pytest
from .. import aggregate_numpy, aggregate_numpy_ufunc, aggregate_purepy
try:
from .. import aggregate_numba
except ImportError:
aggregate_numba = None
try:
from .. import aggregate_pandas
except ImportError:
aggregate_pandas = None
_implementations = [
aggregate_purepy,
aggregate_numpy_ufunc,
aggregate_numpy,
aggregate_numba,
aggregate_pandas,
]
_implementations = [i for i in _implementations if i is not None]
def _impl_name(impl):
if not impl or type(impl).__name__ == "NotSetType":
return
return impl.__name__.rsplit("aggregate_", 1)[1].rsplit("_", 1)[-1]
_not_implemented_by_impl_name = {
"numpy": ("cumprod", "cummax", "cummin"),
"purepy": ("cumsum", "cumprod", "cummax", "cummin", "sumofsquares"),
"numba": ("array", "list", "sort"),
"pandas": ("array", "list", "sort", "sumofsquares", "nansumofsquares"),
"ufunc": "NO_CHECK",
}
def _wrap_notimplemented_skip(impl, name=None):
"""Some implementations lack some functionality. That's ok, let's skip that instead of raising errors."""
@wraps(impl)
def try_skip(*args, **kwargs):
try:
return impl(*args, **kwargs)
except NotImplementedError as e:
impl_name = impl.__module__.split("_")[-1]
func = kwargs.pop("func", None)
if callable(func):
func = func.__name__
not_implemented_ok = _not_implemented_by_impl_name.get(impl_name, [])
if not_implemented_ok == "NO_CHECK" or func in not_implemented_ok:
pytest.skip("Functionality not implemented")
else:
raise e
if name:
try_skip.__name__ = name
return try_skip
func_list = (
"sum",
"prod",
"min",
"max",
"all",
"any",
"mean",
"std",
"var",
"len",
"argmin",
"argmax",
"anynan",
"allnan",
"cumsum",
"sumofsquares",
"nansum",
"nanprod",
"nanmin",
"nanmax",
"nanmean",
"nanstd",
"nanvar",
"nanlen",
"nanargmin",
"nanargmax",
"nansumofsquares",
)
|
