def test_nas():
DT = dt.Frame([d(2001, 5, 17), None, d(2021, 3, 15), None])
RES = DT[:, {"year": year(f[0]), "month": month(f[0]), "day": day(f[0])}]
assert_equals(
RES,
dt.Frame(year=[2001, None, 2021, None],
month=[5, None, 3, None],
day=[17, None, 15, None]))
def test_year_month_day_with_time64_small():
DT = dt.Frame([0, 1, -1, -2, -86400000000000, -86400000000001])
assert DT.type == dt.Type.int64
DT[0] = dt.Type.time64
RES = DT[:, {"year": year(f[0]), "month": month(f[0]), "day": day(f[0])}]
assert_equals(
RES,
dt.Frame(year=[1970, 1970, 1969, 1969, 1969, 1969],
month=[1, 1, 12, 12, 12, 12],
day=[1, 1, 31, 31, 31, 30]))
def test_invalid_type():
DT = dt.Frame(range(5))
msg = r"Function time\.year\(\) requires a date32 or time64 column"
with pytest.raises(TypeError, match=msg):
DT[:, year(f[0])]
msg = r"Function time\.month\(\) requires a date32 or time64 column"
with pytest.raises(TypeError, match=msg):
DT[:, month(f[0])]
msg = r"Function time\.day\(\) requires a date32 or time64 column"
with pytest.raises(TypeError, match=msg):
DT[:, day(f[0])]
def test_noargs():
DT = dt.Frame([None])
msg = r"Function datatable\.year\(\) requires exactly 1 positional argument"
with pytest.raises(TypeError, match=msg):
DT[:, year()]
msg = r"Function datatable\.month\(\) requires exactly 1 positional argument"
with pytest.raises(TypeError, match=msg):
DT[:, month()]
msg = r"Function datatable\.day\(\) requires exactly 1 positional argument"
with pytest.raises(TypeError, match=msg):
DT[:, day()]
def test_year_month_day_with_time64():
DT = dt.Frame([
t(2001, 3, 15, 0, 0, 0),
t(2017, 11, 1, 23, 59, 59, 999999), None,
t(1970, 1, 1, 0, 0, 0, 1),
t(1969, 12, 31, 23, 59, 59, 999999),
t(1900, 12, 13, 0, 0, 0)
])
RES = DT[:, {"year": year(f[0]), "month": month(f[0]), "day": day(f[0])}]
assert_equals(
RES,
dt.Frame(year=[2001, 2017, None, 1970, 1969, 1900],
month=[3, 11, None, 1, 12, 12],
day=[15, 1, None, 1, 31, 13]))
def test_void_column():
DT = dt.Frame([None] * 5)
RES = DT[:, [year(f[0]), month(f[0]), day(f[0])]]
assert_equals(RES, dt.Frame([[None] * 5] * 3))
def test_year_month_day_with_date32():
DT = dt.Frame([d(1970, 1, 2), d(1972, 9, 27), d(2243, 10, 17)])
RES = DT[:, {"year": year(f[0]), "month": month(f[0]), "day": day(f[0])}]
assert_equals(
RES,
dt.Frame(year=[1970, 1972, 2243], month=[1, 9, 10], day=[2, 27, 17]))