def test_gen_buttons_big(self): buttons = gen_buttons([('a', 5), ('b', 20000)], blank=False) buttons = [(b.text, b.callback_data) for b in buttons] assert buttons == [('a 5', 'button:a'), ('b 20k', 'button:b')] buttons = gen_buttons([('a', 5), ('b', 20100)], blank=False) buttons = [(b.text, b.callback_data) for b in buttons] assert buttons == [('a 5', 'button:a'), ('b 20.1k', 'button:b')]
def test_fluid_merge_keyboards(self): bs1 = gen_buttons([('a', 0), ('b', 0)]) bs2 = gen_buttons([('с', 0), ('d', 0)]) kb = fluid_merge_keyboards( InlineKeyboardMarkup([bs1]), None, InlineKeyboardMarkup([bs2]), max_cols=5, ).inline_keyboard assert len(kb) == 1 assert len(kb[0]) == 4
def test_fluid_merge_keyboards_padding(self): bs1 = gen_buttons([('a', 0), ('b', 0)]) bs2 = gen_buttons([('с', 0), ('d', 0)]) kb = fluid_merge_keyboards(InlineKeyboardMarkup([bs1]), None, InlineKeyboardMarkup([bs2]), InlineKeyboardMarkup([bs1]), InlineKeyboardMarkup([bs2]), max_cols=5, padding=True).inline_keyboard assert len(kb) == 2 assert len(kb[0]) == 5 assert len(kb[1]) == 5 assert kb[1][-1].callback_data == EMPTY_CB_DATA
def test_gen_buttons_blank(self): buttons = gen_buttons(['a', ('b', 2)], blank=True) buttons = [(b.text, b.callback_data) for b in buttons] assert buttons == [('a', EMPTY_CB_DATA), ('b 2', EMPTY_CB_DATA)]
def test_gen_buttons(self): buttons = gen_buttons(['a', ('b', 2)], blank=False) buttons = [(b.text, b.callback_data) for b in buttons] assert buttons == [('a', 'button:a'), ('b 2', 'button:b')]