def test_strip(self):
output = TextBuffer()
output.print(" 1 ", format=True)
self.assertEqual([("1\n", True)], output.get_paragraphs())
output.print(" 1 ", format=False)
self.assertEqual([(" 1 \n", False)], output.get_paragraphs())
def test_end(self):
output = TextBuffer()
output.print("1", end=True)
output.print("2", end=True)
self.assertEqual([("1\n", True), ("2\n", True)], output.get_paragraphs())
output.print("one")
output.print("1", end=True)
output.print("two")
output.print("2", end=True)
output.print("three")
self.assertEqual([("one\n1\n", True), ("two\n2\n", True), ("three\n", True)], output.get_paragraphs())
def test_whitespace(self):
output = TextBuffer()
output.print("1")
output.print("2")
output.print("3")
self.assertEqual([("1\n2\n3\n", True)], output.get_paragraphs())
def test_strip(self):
output = TextBuffer()
output.print(" 1 ", format=True)
self.assertEqual([("1\n", True)], output.get_paragraphs())
output.print(" 1 ", format=False)
self.assertEqual([(" 1 \n", False)], output.get_paragraphs())
def test_empty_lines(self):
output = TextBuffer()
output.print("")
output.print("")
self.assertEqual([], output.get_paragraphs(), "empty strings shouldn't be stored")
output.print("", format=False)
output.print("", format=False)
self.assertEqual([("\n\n", False)], output.get_paragraphs(), "2 empty strings without format should be stored in 1 paragraph with 2 new lines")
output.print("", end=True)
output.print("", end=True)
self.assertEqual([("\n", True), ("\n", True)], output.get_paragraphs(), "2 empty strings with end=true should be stored in 2 paragraphs")
output.print("", end=True)
output.print("", end=True)
output.print("", end=True)
self.assertEqual([("\n", True), ("\n", True), ("\n", True)], output.get_paragraphs())
output.print("")
output.print("1")
output.print("2")
output.print("")
self.assertEqual([("1\n2\n", True)], output.get_paragraphs())
def test_whitespace(self):
output = TextBuffer()
output.print("1")
output.print("2")
output.print("3")
self.assertEqual([("1\n2\n3\n", True)], output.get_paragraphs())
def test_end(self):
output = TextBuffer()
output.print("1", end=True)
output.print("2", end=True)
self.assertEqual([("1\n", True), ("2\n", True)],
output.get_paragraphs())
output.print("one")
output.print("1", end=True)
output.print("two")
output.print("2", end=True)
output.print("three")
self.assertEqual([("one\n1\n", True), ("two\n2\n", True),
("three\n", True)], output.get_paragraphs())
def test_empty_lines(self):
output = TextBuffer()
output.print("")
output.print("")
self.assertEqual([], output.get_paragraphs(),
"empty strings shouldn't be stored")
output.print("", format=False)
output.print("", format=False)
self.assertEqual([("\n\n", False)], output.get_paragraphs(
), "2 empty strings without format should be stored in 1 paragraph with 2 new lines"
)
output.print("", end=True)
output.print("", end=True)
self.assertEqual(
[("\n", True), ("\n", True)], output.get_paragraphs(),
"2 empty strings with end=true should be stored in 2 paragraphs")
output.print("", end=True)
output.print("", end=True)
output.print("", end=True)
self.assertEqual([("\n", True), ("\n", True), ("\n", True)],
output.get_paragraphs())
output.print("")
output.print("1")
output.print("2")
output.print("")
self.assertEqual([("1\n2\n", True)], output.get_paragraphs())
def test_text(self):
output = TextBuffer()
output.print("one two three four five six seven")
output.print("eight nine ten eleven twelve thirteen fourteen fifteen")
output.print("sixteen seventeen eighteen nineteen twenty.")
output.p()
output.print("new paragraph.")
output.print("Yeah.", end=True)
output.p()
output.print("new paragraph after empty line.")
output.p()
output.p()
output.print("| x x |", format=False)
output.print("| y y |", format=False)
output.print("| z z |", format=False)
expected = """ one two three four five six seven eight
nine ten eleven twelve thirteen fourteen
fifteen sixteen seventeen eighteen nineteen
twenty.
new paragraph. Yeah.
\x20\x20
new paragraph after empty line.
\x20\x20
| x x |
| y y |
| z z |
"""
io = console_io.ConsoleIo(None)
formatted = io.render_output(output.get_paragraphs(),
indent=2,
width=45)
self.assertEqual(expected, formatted)
def test_text(self):
output = TextBuffer()
output.print("one two three four five six seven")
output.print("eight nine ten eleven twelve thirteen fourteen fifteen")
output.print("sixteen seventeen eighteen nineteen twenty.")
output.p()
output.print("new paragraph.")
output.print("Yeah.", end=True)
output.p()
output.print("new paragraph after empty line.")
output.p()
output.p()
output.print("| x x |", format=False)
output.print("| y y |", format=False)
output.print("| z z |", format=False)
expected = """ one two three four five six seven eight
nine ten eleven twelve thirteen fourteen
fifteen sixteen seventeen eighteen nineteen
twenty.
new paragraph. Yeah.
\x20\x20
new paragraph after empty line.
\x20\x20
| x x |
| y y |
| z z |
"""
io = console_io.ConsoleIo(None)
formatted = io.render_output(output.get_paragraphs(), indent=2, width=45)
self.assertEqual(expected, formatted)