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'