示例#1
0
    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)
示例#2
0
    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)
示例#3
0
    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)