def test_shift_without_domain_shift(): f = Function(np.linspace(0, 10, 100), lambda x: 5 * x) shift = 2 fshifted = Function(f.get_domain(), lambda x: 5 * (x - shift)) f = f.shift(shift) for x in fshifted.get_domain(): assert abs(fshifted(x) - f(x)) == 0
def test_shift_with_domain_shift(): f = Function(np.linspace(0, 10, 100), lambda x: 5 * x) shift = 10 fshifted = Function(np.linspace(10, 20, 100), lambda x: 5 * (x - shift)) f = f.shift(shift, domain=True) for x in fshifted.get_domain(): assert abs(fshifted(x) - f(x)) == 0