def test_roll_ints_with_window_type(self): rolled = rolling(self.int_curve, rolling_window=3, window_type='triang') rolled_vals = [np.NaN, 2, 3, 4, 5, 6, np.NaN] self.assertEqual(rolled, Curve(rolled_vals))
def test_roll_ints(self): rolled = rolling(self.int_curve, rolling_window=2) rolled_vals = [np.NaN, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5] self.assertEqual(rolled, Curve(rolled_vals))