matmul_usecase.py

"""Use cases for testing matmul (@)
"""
def matmul_usecase(x, y):
    return x @ y

def imatmul_usecase(x, y):
    x @= y
    return x

class DumbMatrix(object):

    def __init__(self, value):
        self.value = value

    def __matmul__(self, other):
        if isinstance(other, DumbMatrix):
            return DumbMatrix(self.value * other.value)
        return NotImplemented

    def __imatmul__(self, other):
        if isinstance(other, DumbMatrix):
            self.value *= other.value
            return self
        return NotImplemented