def test_display_display_columns_a(self): 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(i, index=range(4)) self.assertEqual( f.display(config_columns_8).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>' ]) self.assertEqual( f.display(config_columns_5).to_rows(), [ '<FrameGO>', '<IndexGO> ab ... yz <<U2>', '<Index> ...', '0 0 ... 324', '1 0 ... 324', '2 0 ... 324', '3 0 ... 324', '<int64> <int64> ... <int64>' ])
def test_display_cell_fill_width_a(self) -> None: config_width_12 = sf.DisplayConfig.from_default( cell_max_width=12, cell_max_width_leftmost=12, type_color=False) config_width_6 = sf.DisplayConfig.from_default( cell_max_width=6, cell_max_width_leftmost=6, type_color=False) def chunks(size: int, count: int) -> tp.Iterator[str]: pos = 0 for _ in range(count): yield LONG_SAMPLE_STR[pos:pos + size] pos = pos + size s = Series(chunks(20, 3), index=('a', 'b', 'c')) self.assertEqual( s.display(config=config_width_12).to_rows(), [ '<Series>', '<Index>', 'a Lorem ips...', 'b t amet, c...', 'c adipiscin...', '<<U1> <<U20>' ]) self.assertEqual( s.display(config=config_width_6).to_rows(), [ '<Se...', '<In...', 'a Lor...', 'b t a...', 'c adi...', '<<U1> <<U20>' ]) config = sf.DisplayConfig.from_default(type_color=False, cell_max_width_leftmost=20) row_count = 2 index = [str(chr(x)) for x in range(97, 97 + row_count)] f = FrameGO(index=index) for i in range(4): chunker = iter(chunks(10, row_count)) s = Series((x for x in chunker), index=index) f[i] = s f.columns._update_array_cache() self.assertEqual( f.display(config=config).to_rows(), [ '<FrameGO>', '<IndexGO> 0 1 2 3 <int64>', '<Index>', 'a Lorem ipsu Lorem ipsu Lorem ipsu Lorem ipsu', 'b m dolor si m dolor si m dolor si m dolor si', '<<U1> <<U10> <<U10> <<U10> <<U10>' ]) self.assertEqual( f.display(config=config_width_6).to_rows(), [ '<Fr...', '<In... 0 1 2 3 <in...', '<In...', 'a Lor... Lor... Lor... Lor...', 'b m d... m d... m d... m d...', '<<U1> <<U10> <<U10> <<U10> <<U10>' ])