Skip to content
This repository was archived by the owner on Nov 16, 2023. It is now read-only.
10 changes: 10 additions & 0 deletions src/python/tests_extended/test_dft_based.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,8 @@
TEST_CASES = {
'DateTimeSplitter_Simple',
'DateTimeSplitter_Complex',
'DateTimeSplitter_Canada_1day_before_christmas',
'DateTimeSplitter_Czech_non_english_holiday',
'ToKey_SimpleFloat',
'ToKey_SimpleDouble',
'ToKey_SimpleString',
Expand Down Expand Up @@ -62,6 +64,8 @@
'dtAmPmLabel', 'dtDayOfWeekLabel', 'dtIsPaidTimeOff','dtHolidayName'
])
]),
'DateTimeSplitter_Canada_1day_before_christmas' : DateTimeSplitter(prefix='dt', country='Canada') << 'tokens1',
'DateTimeSplitter_Czech_non_english_holiday' : DateTimeSplitter(prefix='dt', country='Czech') << 'tokens1',
'ToKey_SimpleFloat': ToKeyImputer(),
'ToKey_SimpleDouble': ToKeyImputer(),
'ToKey_SimpleString': ToKeyImputer(),
Expand Down Expand Up @@ -131,6 +135,12 @@
'DateTimeSplitter_Complex': pd.DataFrame(data=dict(
tokens1=[217081624, 1751241600, 217081625, 32445842582]
)),
'DateTimeSplitter_Canada_1day_before_christmas': pd.DataFrame(data=dict(
tokens1=[157161599]
)),
'DateTimeSplitter_Czech_non_english_holiday': pd.DataFrame(data=dict(
tokens1=[3911760000, 3834432000, 3985200000]
)),
'ToKey_SimpleFloat': pd.DataFrame(data=dict(
target=[1.0, 1.0, 1.0, 2.0]
)).astype({'target': np.float64}),
Expand Down