def test_heading__level_1(self): result = UI._heading('TEST', level=1) expected_heading = ( '\n\t====================================================' + '\n\t======================[ TEST ]======================' + '\n\t====================================================\n') self.assertEqual(result, expected_heading)
def test_heading__level_3(self): result = UI._heading('TEST', level=3) expected_heading = ( '\n\t----------------------[ TEST ]----------------------\n') self.assertEqual(result, expected_heading)