def test_normalize_window_raises_error_on_window_of_size_zero():
with pytest.raises(ValueError):
x = ts.generate_series(10)
normalize_window(x, 0)
with pytest.raises(ValueError):
x = ts.generate_series(10)
normalize_window(x, Window(0, 0))
def test_normalize_window_single_str():
x = ts.generate_series(10)
w = normalize_window(x, "2d")
assert w.w == pd.DateOffset(days=2)
assert w.r == pd.DateOffset(days=2)
x = ts.generate_series(90)
w = normalize_window(x, "3m")
assert w.w == pd.DateOffset(months=3)
assert w.r == pd.DateOffset(months=3)
def test_normalize_window_handles_int():
x = ts.generate_series(10)
w = normalize_window(x, 5)
assert w.w == 5
assert w.r == 5
def test_normalize_window_defaults_window_if_passed():
x = ts.generate_series(10)
w = normalize_window(x, None, default_window=2)
assert w.w == 2
assert w.r == 0
def test_normalize_window_defaults_window_if_none_passed():
x = ts.generate_series(10)
w = normalize_window(x, None)
assert w.w == 10
assert w.r == 0
def test_normalize_window_str():
x = ts.generate_series(10)
w = normalize_window(x, Window('1w', '2d'))
assert w.w == pd.DateOffset(weeks=1)
assert w.r == pd.DateOffset(days=2)
def test_normalize_window_handles_ramp_of_size_zero():
x = ts.generate_series(10)
w = normalize_window(x, Window(2, 0))
assert w.w == 2
assert w.r == 0
def test_normalize_window_handles_ramp_greater_than_series_length():
with pytest.raises(ValueError):
x = ts.generate_series(10)
normalize_window(x, Window(2, 11))
def test_normalize_window_handles_window_with_no_size():
x = ts.generate_series(10)
w = normalize_window(x, Window(None, 2))
assert w.w == 10
assert w.r == 2
def test_normalize_window_handles_window_with_no_ramp():
x = ts.generate_series(10)
w = normalize_window(x, Window(2, None))
assert w.w == 2
assert w.r == 2
