def test_shifting():
# Kernels:
yield eq, str(ZeroKernel().shift(5)), '0'
yield eq, str(EQ().shift(5)), 'EQ()'
yield eq, str(Linear().shift(5)), 'Linear() shift 5'
yield eq, str((5 * EQ()).shift(5)), '5 * EQ()'
yield eq, str((5 * Linear()).shift(5)), '(5 * Linear()) shift 5'
# Means:
def mean(x):
return x
m = TensorProductMean(mean)
yield eq, str(ZeroMean().shift(5)), '0'
yield eq, str(m.shift(5)), 'mean shift 5'
yield eq, str(m.shift(5).shift(5)), 'mean shift 10'
yield eq, str((5 * m).shift(5)), '(5 * mean) shift 5'
def test_shifting():
# Kernels:
assert str(ZeroKernel().shift(5)) == '0'
assert str(EQ().shift(5)) == 'EQ()'
assert str(Linear().shift(5)) == 'Linear() shift 5'
assert str((5 * EQ()).shift(5)) == '5 * EQ()'
assert str((5 * Linear()).shift(5)) == '(5 * Linear()) shift 5'
# Means:
def mean(x):
return x
m = TensorProductMean(mean)
assert str(ZeroMean().shift(5)) == '0'
assert str(m.shift(5)) == 'mean shift 5'
assert str(m.shift(5).shift(5)) == 'mean shift 10'
assert str((5 * m).shift(5)) == '(5 * mean) shift 5'