def test_to_main_no_main(self): b = f''' '''.strip().splitlines() expected_result = f''' {DECK_MAIN} 1 Embercleave '''.strip().splitlines() actual_result = add_to_section(b, 'Embercleave', 1, 1) self.assertEqual(actual_result, expected_result)
def test_to_other_with_count_two(self): b = f''' {DECK_OTHER} '''.lstrip().splitlines() expected_result = f''' {DECK_OTHER} 2 Embercleave '''.lstrip().splitlines() actual_result = add_to_section(b, 'Embercleave', 2, 3) self.assertEqual(actual_result, expected_result)
def test_to_new_sb_with_main_and_other(self): b = f''' {DECK_MAIN} {DECK_OTHER} --- '''.lstrip().splitlines() expected_result = f''' {DECK_MAIN} {DECK_SB} 1 Embercleave {DECK_OTHER} --- '''.lstrip().splitlines() actual_result = add_to_section(b, 'Embercleave', 1, 2) self.assertEqual(actual_result, expected_result)