def test_shift_serie(): """Test the routine for shifting.""" # Create a synthetic signal signal = np.arange(5) # Create a shift of 0 shift = 0 signal_shift = StandardTimeNormalization._shift_serie(signal, shift) # Check the signal assert_array_equal(signal_shift, signal) # Create a shift of 2 shift = 2 signal_shift = StandardTimeNormalization._shift_serie(signal, shift) # Check the signal gt_signal = np.array([0, 0, 0, 1, 2]) assert_array_equal(signal_shift, gt_signal) # Create a shift of -2 shift = -2 signal_shift = StandardTimeNormalization._shift_serie(signal, shift) # Check the signal gt_signal = np.array([2, 3, 4, 4, 4]) assert_array_equal(signal_shift, gt_signal)