def test_dti_snap(name, tz):
dti = DatetimeIndex([
'1/1/2002', '1/2/2002', '1/3/2002', '1/4/2002', '1/5/2002', '1/6/2002',
'1/7/2002'
],
name=name,
tz=tz,
freq='D')
result = dti.snap(freq='W-MON')
expected = date_range('12/31/2001',
'1/7/2002',
name=name,
tz=tz,
freq='w-mon')
expected = expected.repeat([3, 4])
tm.assert_index_equal(result, expected)
assert result.tz == expected.tz
result = dti.snap(freq='B')
expected = date_range('1/1/2002', '1/7/2002', name=name, tz=tz, freq='b')
expected = expected.repeat([1, 1, 1, 2, 2])
tm.assert_index_equal(result, expected)
assert result.tz == expected.tz
def test_dti_snap(name, tz):
dti = DatetimeIndex(
[
"1/1/2002",
"1/2/2002",
"1/3/2002",
"1/4/2002",
"1/5/2002",
"1/6/2002",
"1/7/2002",
],
name=name,
tz=tz,
freq="D",
)
result = dti.snap(freq="W-MON")
expected = date_range("12/31/2001",
"1/7/2002",
name=name,
tz=tz,
freq="w-mon")
expected = expected.repeat([3, 4])
tm.assert_index_equal(result, expected)
assert result.tz == expected.tz
result = dti.snap(freq="B")
expected = date_range("1/1/2002", "1/7/2002", name=name, tz=tz, freq="b")
expected = expected.repeat([1, 1, 1, 2, 2])
tm.assert_index_equal(result, expected)
assert result.tz == expected.tz
def test_dti_snap():
dti = DatetimeIndex(['1/1/2002', '1/2/2002', '1/3/2002', '1/4/2002',
'1/5/2002', '1/6/2002', '1/7/2002'], freq='D')
res = dti.snap(freq='W-MON')
exp = date_range('12/31/2001', '1/7/2002', freq='w-mon')
exp = exp.repeat([3, 4])
assert (res == exp).all()
res = dti.snap(freq='B')
exp = date_range('1/1/2002', '1/7/2002', freq='b')
exp = exp.repeat([1, 1, 1, 2, 2])
assert (res == exp).all()
def test_dti_snap():
dti = DatetimeIndex(['1/1/2002', '1/2/2002', '1/3/2002', '1/4/2002',
'1/5/2002', '1/6/2002', '1/7/2002'], freq='D')
res = dti.snap(freq='W-MON')
exp = date_range('12/31/2001', '1/7/2002', freq='w-mon')
exp = exp.repeat([3, 4])
assert (res == exp).all()
res = dti.snap(freq='B')
exp = date_range('1/1/2002', '1/7/2002', freq='b')
exp = exp.repeat([1, 1, 1, 2, 2])
assert (res == exp).all()
def test_dti_snap(self):
dti = DatetimeIndex(['1/1/2002', '1/2/2002', '1/3/2002', '1/4/2002',
'1/5/2002', '1/6/2002', '1/7/2002'], freq='D')
res = dti.snap(freq='W-MON')
exp = DatetimeIndex(['12/31/2001', '12/31/2001', '12/31/2001',
'1/7/2002', '1/7/2002', '1/7/2002', '1/7/2002'],
freq='W-MON')
self.assert_( (res == exp).all() )
res = dti.snap(freq='B')
exp = DatetimeIndex(['1/1/2002', '1/2/2002', '1/3/2002', '1/4/2002',
'1/4/2002', '1/7/2002', '1/7/2002'], freq='B')
self.assert_( (res == exp).all() )
def test_dti_snap(name, tz):
dti = DatetimeIndex(['1/1/2002', '1/2/2002', '1/3/2002', '1/4/2002',
'1/5/2002', '1/6/2002', '1/7/2002'],
name=name, tz=tz, freq='D')
result = dti.snap(freq='W-MON')
expected = date_range('12/31/2001', '1/7/2002',
name=name, tz=tz, freq='w-mon')
expected = expected.repeat([3, 4])
tm.assert_index_equal(result, expected)
assert result.tz == expected.tz
result = dti.snap(freq='B')
expected = date_range('1/1/2002', '1/7/2002',
name=name, tz=tz, freq='b')
expected = expected.repeat([1, 1, 1, 2, 2])
tm.assert_index_equal(result, expected)
assert result.tz == expected.tz
def test_dti_snap(self):
dti = DatetimeIndex(
["1/1/2002", "1/2/2002", "1/3/2002", "1/4/2002", "1/5/2002", "1/6/2002", "1/7/2002"], freq="D"
)
res = dti.snap(freq="W-MON")
exp = DatetimeIndex(
["12/31/2001", "12/31/2001", "12/31/2001", "1/7/2002", "1/7/2002", "1/7/2002", "1/7/2002"], freq="W-MON"
)
self.assert_((res == exp).all())
res = dti.snap(freq="B")
exp = DatetimeIndex(
["1/1/2002", "1/2/2002", "1/3/2002", "1/4/2002", "1/4/2002", "1/7/2002", "1/7/2002"], freq="B"
)
self.assert_((res == exp).all())
def test_dti_snap(name, tz, unit):
dti = DatetimeIndex(
[
"1/1/2002",
"1/2/2002",
"1/3/2002",
"1/4/2002",
"1/5/2002",
"1/6/2002",
"1/7/2002",
],
name=name,
tz=tz,
freq="D",
)
dti = astype_non_nano(dti, unit)
result = dti.snap(freq="W-MON")
expected = date_range("12/31/2001",
"1/7/2002",
name=name,
tz=tz,
freq="w-mon")
expected = expected.repeat([3, 4])
expected = astype_non_nano(expected, unit)
tm.assert_index_equal(result, expected)
assert result.tz == expected.tz
assert result.freq is None
assert expected.freq is None
result = dti.snap(freq="B")
expected = date_range("1/1/2002", "1/7/2002", name=name, tz=tz, freq="b")
expected = expected.repeat([1, 1, 1, 2, 2])
expected = astype_non_nano(expected, unit)
tm.assert_index_equal(result, expected)
assert result.tz == expected.tz
assert result.freq is None
assert expected.freq is None
