2024817581

Build # 2024817581

Build Type

github

Committed by GitHub

Commit Message

Merge 6b9e6b1b2 into f68237ee3

Pull Request Pull Request #187: Add `gb.agg.from_string`

Run Details

3918 of 4330 branches covered (90.48%)

Branch coverage included in aggregate %.

35 of 35 new or added lines in 5 files covered. (100.0%)

1175 existing lines in 21 files now uncovered.

12596 of 13771 relevant lines covered (91.47%)

13.7 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

87.23

/grblas/binary/numpy.py

""" Create UDFs of numpy functions supported by numba.

See list of numpy ufuncs supported by numpy here:

https://numba.pydata.org/numba-doc/dev/reference/numpysupported.html#math-operations

"""
import numpy as _np

from .. import binary as _binary
from .. import config as _config
from .. import operator as _operator

_delayed = {}
_binary_names = {
    # Math operations
    "add",
    "subtract",
    "multiply",
    "divide",
    "logaddexp",
    "logaddexp2",
    "true_divide",
    "floor_divide",
    "power",
    "remainder",
    "mod",
    "fmod",
    "gcd",
    "lcm",
    # Trigonometric functions
    "arctan2",
    "hypot",
    # Bit-twiddling functions
    "bitwise_and",
    "bitwise_or",
    "bitwise_xor",
    "left_shift",
    "right_shift",
    # Comparison functions
    "greater",
    "greater_equal",
    "less",
    "less_equal",
    "not_equal",
    "equal",
    "logical_and",
    "logical_or",
    "logical_xor",
    "maximum",
    "minimum",
    "fmax",
    "fmin",
    # Floating functions
    "copysign",
    "nextafter",
    "ldexp",
}
_operator._STANDARD_OPERATOR_NAMES.update(f"binary.numpy.{name}" for name in _binary_names)
__all__ = list(_binary_names)
_numpy_to_graphblas = {
    # Monoids
    "add": "plus",
    "bitwise_and": "band",
    "bitwise_or": "bor",
    "bitwise_xor": "bxor",
    "equal": "eq",
    "fmax": "max",  # ignores nan
    "fmin": "min",  # ignores nan
    "logical_and": "land",
    "logical_or": "lor",
    "logical_xor": "lxor",
    "multiply": "times",
    # Other
    "arctan2": "atan2",
    "copysign": "copysign",
    "divide": "truediv",
    # "floor_divide": "floordiv",  # floor_divide does not cast to int, but floordiv does
    # "fmod": "fmod",  # not the same!
    "greater": "gt",
    "greater_equal": "ge",
    "ldexp": "ldexp",
    "less": "lt",
    "less_equal": "le",
    # "mod": "remainder",  # not the same!
    "not_equal": "ne",
    "power": "pow",
    # "remainder": "remainder",  # not the same!
    "subtract": "minus",
    "true_divide": "truediv",
}
# Not included: maximum, minimum, gcd, hypot, logaddexp, logaddexp2
# lcm, left_shift, nextafter, right_shift

_commutative = {
    "add",
    "bitwise_and",
    "bitwise_or",
    "bitwise_xor",
    "equal",
    "fmax",
    "fmin",
    "gcd",
    "hypot",
    "lcm",
    "logaddexp",
    "logaddexp2",
    "logical_and",
    "logical_or",
    "logical_xor",
    "maximum",
    "minimum",
    "multiply",
    "not_equal",
}
_commutes_to = {
    "greater": "less",
    "greater_equal": "less_equal",
    "less": "greater",
    "less_equal": "greater_equal",
}
# Don't commute: arctan2, copysign, divide, floor_divide, fmod, ldexp, left_shift,
# mod, nextafter, power, remainder, right_shift, subtract, true_divide.
# If desired, we could create r-versions of these so they can commute to something.


def __dir__():
    return __all__ + list(_delayed)


def __getattr__(name):
    if name in _delayed:
        func, kwargs = _delayed.pop(name)
        rv = func(**kwargs)
        globals()[name] = rv
        return rv
    if name not in _binary_names:
        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
    if _config.get("mapnumpy") and name in _numpy_to_graphblas:
        globals()[name] = getattr(_binary, _numpy_to_graphblas[name])
    else:
        numpy_func = getattr(_np, name)
        _operator.BinaryOp.register_new(f"numpy.{name}", lambda x, y: numpy_func(x, y))
    rv = globals()[name]
    if name in _commutative:
        rv.commutes_to = rv
    elif name in _commutes_to:
        right = getattr(_binary.numpy, _commutes_to[name])
        rv.commutes_to = rv.commutes_to or right
        right.commutes_to = right.commutes_to or rv
    return rv

1	""" Create UDFs of numpy functions supported by numba.
2
3	See list of numpy ufuncs supported by numpy here:
4
5	https://numba.pydata.org/numba-doc/dev/reference/numpysupported.html#math-operations
6
7	"""
8	import numpy as _np	15✔
9
10	from .. import binary as _binary	15✔
11	from .. import config as _config	15✔
12	from .. import operator as _operator	15✔
13
14	_delayed = {}	15✔
15	_binary_names = {	15✔
16	# Math operations
17	"add",
18	"subtract",
19	"multiply",
20	"divide",
21	"logaddexp",
22	"logaddexp2",
23	"true_divide",
24	"floor_divide",
25	"power",
26	"remainder",
27	"mod",
28	"fmod",
29	"gcd",
30	"lcm",
31	# Trigonometric functions
32	"arctan2",
33	"hypot",
34	# Bit-twiddling functions
35	"bitwise_and",
36	"bitwise_or",
37	"bitwise_xor",
38	"left_shift",
39	"right_shift",
40	# Comparison functions
41	"greater",
42	"greater_equal",
43	"less",
44	"less_equal",
45	"not_equal",
46	"equal",
47	"logical_and",
48	"logical_or",
49	"logical_xor",
50	"maximum",
51	"minimum",
52	"fmax",
53	"fmin",
54	# Floating functions
55	"copysign",
56	"nextafter",
57	"ldexp",
58	}
59	_operator._STANDARD_OPERATOR_NAMES.update(f"binary.numpy.{name}" for name in _binary_names)	15✔
60	__all__ = list(_binary_names)	15✔
61	_numpy_to_graphblas = {	15✔
62	# Monoids
63	"add": "plus",
64	"bitwise_and": "band",
65	"bitwise_or": "bor",
66	"bitwise_xor": "bxor",
67	"equal": "eq",
68	"fmax": "max", # ignores nan
69	"fmin": "min", # ignores nan
70	"logical_and": "land",
71	"logical_or": "lor",
72	"logical_xor": "lxor",
73	"multiply": "times",
74	# Other
75	"arctan2": "atan2",
76	"copysign": "copysign",
77	"divide": "truediv",
78	# "floor_divide": "floordiv", # floor_divide does not cast to int, but floordiv does
79	# "fmod": "fmod", # not the same!
80	"greater": "gt",
81	"greater_equal": "ge",
82	"ldexp": "ldexp",
83	"less": "lt",
84	"less_equal": "le",
85	# "mod": "remainder", # not the same!
86	"not_equal": "ne",
87	"power": "pow",
88	# "remainder": "remainder", # not the same!
89	"subtract": "minus",
90	"true_divide": "truediv",
91	}
92	# Not included: maximum, minimum, gcd, hypot, logaddexp, logaddexp2
93	# lcm, left_shift, nextafter, right_shift
94
95	_commutative = {	15✔
96	"add",
97	"bitwise_and",
98	"bitwise_or",
99	"bitwise_xor",
100	"equal",
101	"fmax",
102	"fmin",
103	"gcd",
104	"hypot",
105	"lcm",
106	"logaddexp",
107	"logaddexp2",
108	"logical_and",
109	"logical_or",
110	"logical_xor",
111	"maximum",
112	"minimum",
113	"multiply",
114	"not_equal",
115	}
116	_commutes_to = {	15✔
117	"greater": "less",
118	"greater_equal": "less_equal",
119	"less": "greater",
120	"less_equal": "greater_equal",
121	}
122	# Don't commute: arctan2, copysign, divide, floor_divide, fmod, ldexp, left_shift,
123	# mod, nextafter, power, remainder, right_shift, subtract, true_divide.
124	# If desired, we could create r-versions of these so they can commute to something.
125
126
127	def __dir__():	15✔
128	return __all__ + list(_delayed)	15✔
129
130
131	def __getattr__(name):	15✔
132	if name in _delayed:	15✔
133	func, kwargs = _delayed.pop(name)	15✔
134	rv = func(**kwargs)	15✔
135	globals()[name] = rv	15✔
136	return rv	15✔
137	if name not in _binary_names:	15✔
138	raise AttributeError(f"module {__name__!r} has no attribute {name!r}")	15✔
139	if _config.get("mapnumpy") and name in _numpy_to_graphblas:	15✔
140	globals()[name] = getattr(_binary, _numpy_to_graphblas[name])	15✔
141	else:
142	numpy_func = getattr(_np, name)	15✔
143	_operator.BinaryOp.register_new(f"numpy.{name}", lambda x, y: numpy_func(x, y))	15✔
144	rv = globals()[name]	15✔
145	if name in _commutative:	15!
UNCOV 146	rv.commutes_to = rv	×
147	elif name in _commutes_to:	15!
UNCOV 148	right = getattr(_binary.numpy, _commutes_to[name])	×
UNCOV 149	rv.commutes_to = rv.commutes_to or right	×
UNCOV 150	right.commutes_to = right.commutes_to or rv	×
151	return rv	15✔

metagraph-dev / grblas / 2024817581

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous