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)
