def test_index_date_threshold_a(self) -> None:
index = IndexDate.from_date_range('2019-01-01', '2020-02-28')
threshold_day = 15
threshold_month = 2
lag = 2
f = FrameGO(index=index)
f['month_day'] = f.index.via_dt.month * 100 + f.index.via_dt.day
f['transition'] = f['month_day'] >= (threshold_month * 100 +
threshold_day)
f['year_shift'] = Series.from_element(
lag, index=index).assign[f['transition']](lag - 1)
f['year'] = f.index.via_dt.year - f['year_shift'].values
self.assertEqual(f['transition'].sum(), 334)
self.assertEqual(f['year'].values.tolist(), [
2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017,
2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017,
2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017,
2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017, 2017,
2017, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018, 2018,
2018, 2018, 2018, 2019, 2019, 2019, 2019, 2019, 2019, 2019, 2019,
2019, 2019, 2019, 2019, 2019, 2019
])
def test_display_display_columns_a(self) -> None:
config_columns_8 = sf.DisplayConfig.from_default(display_columns=8,
type_color=False)
config_columns_5 = sf.DisplayConfig.from_default(display_columns=5,
type_color=False)
columns = list(''.join(x)
for x in combinations(string.ascii_lowercase, 2))
f = FrameGO(index=range(4))
for i, col in enumerate(columns):
f[col] = Series.from_element(i, index=range(4))
self.assertEqual(
f.display(config_columns_8).to_rows(), [
'<FrameGO>',
'<IndexGO> ab ac ad ... xy xz yz <<U2>',
'<Index> ...',
'0 0 1 2 ... 322 323 324',
'1 0 1 2 ... 322 323 324',
'2 0 1 2 ... 322 323 324',
'3 0 1 2 ... 322 323 324',
'<int64> <int64> <int64> <int64> ... <int64> <int64> <int64>'
])
self.assertEqual(
f.display(config_columns_5).to_rows(), [
'<FrameGO>',
'<IndexGO> ab ac ... xz yz <<U2>',
'<Index> ...',
'0 0 1 ... 323 324',
'1 0 1 ... 323 324',
'2 0 1 ... 323 324',
'3 0 1 ... 323 324',
'<int64> <int64> <int64> ... <int64> <int64>'
])