2163383767

Build Type

github

Committed by GitHub

Commit Message

Merge 2439109e3 into ef6e61a57

Pull Request Pull Request #207: Add `select` Operation

Run Details

4729 of 4764 branches covered (99.27%)

Branch coverage included in aggregate %.

154 of 175 new or added lines in 8 files covered. (88.0%)

1 existing line in 1 file now uncovered.

15129 of 15160 relevant lines covered (99.8%)

14.22 hits per line

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

66.67

/graphblas/select/__init__.py

# All items are dynamically added by classes in operator.py
# This module acts as a container of SelectOp instances (which are
# identical to IndexUnaryOp under the hood, except they must return
# boolean and they default to `select` when called, rather than
# `apply`).
_delayed = {}
from graphblas import operator  # noqa isort:skip

del operator


def __dir__():
    return globals().keys() | _delayed.keys()


def __getattr__(key):
    if key in _delayed:
        func, kwargs = _delayed.pop(key)
        rv = func(**kwargs)
        globals()[key] = rv
        return rv
    raise AttributeError(f"module {__name__!r} has no attribute {key!r}")


def _resolve_expr(expr, callname, opname):
    from graphblas.base import BaseExpression

    if not isinstance(expr, BaseExpression) and hasattr(expr, "select"):
        raise TypeError(
            f"Expected VectorExpression or MatrixExpression; found {type(expr)}\n"
            f"Typical usage: select.{callname}(x <= 5)"
        )
    tensor = expr.args[0]
    thunk = expr.args[1]
    method = f"{opname}{expr.op.name}"
    if method not in globals():
        # Convert thunk to Python int to avoid possible subtraction with uints
        thunk = int(thunk.value)
        # Attempt to convert < into <= (rowlt is not part of official spec, but rowle is)
        if expr.op.name == "lt":
            method = f"{opname}le"
            thunk -= 1
        # Attempt to convert >= into > (rowge is not part of official spec, but rowgt is)
        elif expr.op.name == "ge":
            method = f"{opname}gt"
            thunk -= 1
        if method not in globals():
            raise ValueError(f"Unknown or unregistered select method: {method}")
    return globals()[method](tensor, thunk)


def value(expr):
    return _resolve_expr(expr, "value", "value")


def row(expr):
    return _resolve_expr(expr, "row", "row")


def column(expr):
    return _resolve_expr(expr, "column", "col")

1	# All items are dynamically added by classes in operator.py
2	# This module acts as a container of SelectOp instances (which are
3	# identical to IndexUnaryOp under the hood, except they must return
4	# boolean and they default to `select` when called, rather than
5	# `apply`).
6	_delayed = {}	15✔
7	from graphblas import operator # noqa isort:skip	15✔
8
9	del operator	15✔
10
11
12	def __dir__():	15✔
NEW 13	return globals().keys() \| _delayed.keys()	×
14
15
16	def __getattr__(key):	15✔
17	if key in _delayed:	15!
NEW 18	func, kwargs = _delayed.pop(key)	×
NEW 19	rv = func(**kwargs)	×
NEW 20	globals()[key] = rv	×
NEW 21	return rv	×
22	raise AttributeError(f"module {__name__!r} has no attribute {key!r}")	15✔
23
24
25	def _resolve_expr(expr, callname, opname):	15✔
26	from graphblas.base import BaseExpression	15✔
27
28	if not isinstance(expr, BaseExpression) and hasattr(expr, "select"):	15!
NEW 29	raise TypeError(	×
30	f"Expected VectorExpression or MatrixExpression; found {type(expr)}\n"
31	f"Typical usage: select.{callname}(x <= 5)"
32	)
33	tensor = expr.args[0]	15✔
34	thunk = expr.args[1]	15✔
35	method = f"{opname}{expr.op.name}"	15✔
36	if method not in globals():	15✔
37	# Convert thunk to Python int to avoid possible subtraction with uints
38	thunk = int(thunk.value)	15✔
39	# Attempt to convert < into <= (rowlt is not part of official spec, but rowle is)
40	if expr.op.name == "lt":	15!
41	method = f"{opname}le"	15✔
42	thunk -= 1	15✔
43	# Attempt to convert >= into > (rowge is not part of official spec, but rowgt is)
NEW 44	elif expr.op.name == "ge":	×
NEW 45	method = f"{opname}gt"	×
NEW 46	thunk -= 1	×
47	if method not in globals():	15!
NEW 48	raise ValueError(f"Unknown or unregistered select method: {method}")	×
49	return globals()[method](tensor, thunk)	15✔
50
51
52	def value(expr):	15✔
53	return _resolve_expr(expr, "value", "value")	15✔
54
55
56	def row(expr):	15✔
57	return _resolve_expr(expr, "row", "row")	15✔
58
59
60	def column(expr):	15✔
61	return _resolve_expr(expr, "column", "col")	15✔

metagraph-dev / grblas / 2163383767

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