def test_one_line_string(self):
def error_msg(str_odd=True, width_odd=True):
string = "Odd" if str_odd else "Even"
width = "odd" if width_odd else "even"
return "%s string on %s width centered incorrectly." % (string,
width)
self.assertEquals(center_text("AAAA", 11),
" AAAA ".rstrip(),
msg=error_msg(True, False))
self.assertEquals(center_text("AAA", 11),
" AAA ".rstrip(),
msg=error_msg(False, False))
self.assertEquals(center_text("AA", 11),
" AA ".rstrip(),
msg=error_msg(True, False))
self.assertEquals(center_text("AAAA", 10),
" AAAA ".rstrip(),
msg=error_msg(True, True))
self.assertEquals(center_text("AAA", 10),
" AAA ".rstrip(),
msg=error_msg(False, True))
self.assertEquals(center_text("AA", 10),
" AA ".rstrip(),
msg=error_msg(True, True))
def test_multiple_lines_string(self):
self.assertEquals(center_text("AAA\nBBBBB", 7), " AAA\n BBBBB")
def test_string_longer_than_width(self):
self.assertEquals(center_text("#" * 10, 5), "#" * 10)
def test_empty_string(self):
self.assertEquals(center_text("", 10), " " * 5)
