def test_corner_cases():
with pytest.raises(IndexError):
EQ().stretch(1)[1]
with pytest.raises(IndexError):
(EQ() + RQ(1))[2]
with pytest.raises(RuntimeError):
mul(1, 1)
with pytest.raises(RuntimeError):
add(1, 1)
with pytest.raises(RuntimeError):
stretch(1, 1)
with pytest.raises(RuntimeError):
transform(1, 1)
with pytest.raises(RuntimeError):
differentiate(1, 1)
with pytest.raises(RuntimeError):
select(1, 1)
with pytest.raises(RuntimeError):
shift(1, 1)
assert repr(EQ()) == str(EQ())
assert EQ().__name__ == 'EQ'
with pytest.raises(NotImplementedError):
WrappedElement(1).display(1, lambda x: x)
with pytest.raises(NotImplementedError):
JoinElement(1, 2).display(1, 2, lambda x: x)
def test_corner_cases():
yield raises, IndexError, lambda: EQ().stretch(1)[1]
yield raises, IndexError, lambda: (EQ() + RQ(1))[2]
yield raises, RuntimeError, lambda: mul(1, 1)
yield raises, RuntimeError, lambda: add(1, 1)
yield raises, RuntimeError, lambda: stretch(1, 1)
yield raises, RuntimeError, lambda: transform(1, 1)
yield raises, RuntimeError, lambda: differentiate(1, 1)
yield raises, RuntimeError, lambda: select(1, 1)
yield raises, RuntimeError, lambda: shift(1, 1)
yield eq, repr(EQ()), str(EQ())
yield eq, EQ().__name__, 'EQ'
yield raises, NotImplementedError, \
lambda: WrappedElement(1).display(1, lambda x: x)
yield raises, NotImplementedError, \
lambda: JoinElement(1, 2).display(1, 2, lambda x: x)
def mul(a, b):
return mul(new(b, TensorProductFunction)(a), b)
def mul(a, b):
return mul(a, new(a, TensorProductFunction)(b))
def mul(a, b): return mul(new(b, TensorProductFunction)(a), b) @add.extend(Element, PythonFunction, precedence=10)
def mul(a, b): return mul(a, new(a, TensorProductFunction)(b)) @mul.extend(PythonFunction, Element, precedence=10)
