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 reverse(a): return stretch(reversed(a[0]), *reversed(a.stretches))