def test_typed(): class A(Basic): pass class B(Basic): pass rmzeros = rm_id(lambda x: x == S(0)) rmones = rm_id(lambda x: x == S(1)) remove_something = typed({A: rmzeros, B: rmones}) assert remove_something(A(S(0), S(1))) == A(S(1)) assert remove_something(B(S(0), S(1))) == B(S(0))
def test_typed(): class A(Basic): pass class B(Basic): pass rmzeros = rm_id(lambda x: x == 0) rmones = rm_id(lambda x: x == 1) remove_something = typed({A: rmzeros, B: rmones}) assert remove_something(A(0, 1)) == A(1) assert remove_something(B(0, 1)) == B(0)
def test_rm_id(): rmzeros = rm_id(lambda x: x == 0) assert rmzeros(Basic(0, 1)) == Basic(1) assert rmzeros(Basic(0, 0)) == Basic(0) assert rmzeros(Basic(2, 1)) == Basic(2, 1)
def test_rm_id(): rmzeros = rm_id(lambda x: x == 0) assert rmzeros(Basic(S(0), S(1))) == Basic(S(1)) assert rmzeros(Basic(S(0), S(0))) == Basic(S(0)) assert rmzeros(Basic(S(2), S(1))) == Basic(S(2), S(1))