1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
import unittest2
from StringIO import StringIO
from screenplain.export.fdx import write_text
from screenplain.richstring import plain, bold, italic
class OutputTests(unittest2.TestCase):
def setUp(self):
self.out = StringIO()
def test_plain_text_should_have_no_style(self):
write_text(self.out, plain('hello'), False)
self.assertEqual(
self.out.getvalue(),
' <Text>hello</Text>\n'
)
def test_bold_text_should_have_bold_style(self):
write_text(self.out, bold('hello'), False)
self.assertEqual(
self.out.getvalue(),
' <Text style="Bold">hello</Text>\n'
)
def test_sequential_styles(self):
rich = plain('plain') + bold('b') + italic('i')
write_text(self.out, rich, False)
self.assertEqual(
self.out.getvalue(),
' <Text>plain</Text>\n'
' <Text style="Bold">b</Text>\n'
' <Text style="Italic">i</Text>\n'
)
def test_several_styles(self):
rich = bold('outer') + (bold + italic)('inner') + bold('outer')
write_text(self.out, rich, False)
self.assertEqual(
self.out.getvalue(),
' <Text style="Bold">outer</Text>\n'
' <Text style="Bold+Italic">inner</Text>\n'
' <Text style="Bold">outer</Text>\n'
)
def test_write_text_adds_line_break_if_requested(self):
rich = bold('outer') + (bold + italic)('inner') + bold('outer')
write_text(self.out, rich, True)
self.assertEqual(
self.out.getvalue(),
' <Text style="Bold">outer</Text>\n'
' <Text style="Bold+Italic">inner</Text>\n'
' <Text style="Bold">outer\n</Text>\n' # note newline
)
|
