~ubuntu-branches/ubuntu/trusty/pexpect/trusty-proposed

« back to all changes in this revision

Viewing changes to tests/test_screen.py

  • Committer: Package Import Robot
  • Author(s): Andrew Starr-Bochicchio, Thomas Kluyver, Jakub Wilk, Jackson Doak, Andrew Starr-Bochicchio
  • Date: 2013-12-06 20:20:26 UTC
  • mfrom: (1.1.4)
  • Revision ID: package-import@ubuntu.com-20131206202026-9k9oixbv7e8ke30q
Tags: 3.0-1
* Team upload.

[ Thomas Kluyver ]
* New upstream release. Closes: #729518
* Add packaging for Python 3.
* Use pybuild for packaging.

[ Jakub Wilk ]
* Use canonical URIs for Vcs-* fields.

[ Jackson Doak ]
* Create debian/python3-pexpect.docs

[ Andrew Starr-Bochicchio ]
* Remove empty debian/patches dir.
* Move documentation and examples into a new python-pexpect-doc
  package. They are shared between the Python 2 and Python 3 packages,
  so there is no need to install them with both.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#!/usr/bin/env python
 
2
'''
 
3
PEXPECT LICENSE
 
4
 
 
5
    This license is approved by the OSI and FSF as GPL-compatible.
 
6
        http://opensource.org/licenses/isc-license.txt
 
7
 
 
8
    Copyright (c) 2012, Noah Spurrier <noah@noah.org>
 
9
    PERMISSION TO USE, COPY, MODIFY, AND/OR DISTRIBUTE THIS SOFTWARE FOR ANY
 
10
    PURPOSE WITH OR WITHOUT FEE IS HEREBY GRANTED, PROVIDED THAT THE ABOVE
 
11
    COPYRIGHT NOTICE AND THIS PERMISSION NOTICE APPEAR IN ALL COPIES.
 
12
    THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 
13
    WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 
14
    MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 
15
    ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 
16
    WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 
17
    ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 
18
    OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 
19
 
 
20
'''
 
21
 
 
22
from pexpect import screen
 
23
import unittest
 
24
import PexpectTestCase
 
25
 
 
26
fill1_target='XXXXXXXXXX\n' + \
 
27
'XOOOOOOOOX\n' + \
 
28
'XO::::::OX\n' + \
 
29
'XO:oooo:OX\n' + \
 
30
'XO:o..o:OX\n' + \
 
31
'XO:o..o:OX\n' + \
 
32
'XO:oooo:OX\n' + \
 
33
'XO::::::OX\n' + \
 
34
'XOOOOOOOOX\n' + \
 
35
'XXXXXXXXXX'
 
36
fill2_target = 'XXXXXXXXXXX\n' + \
 
37
'XOOOOOOOOOX\n' + \
 
38
'XO:::::::OX\n' + \
 
39
'XO:ooooo:OX\n' + \
 
40
'XO:o...o:OX\n' + \
 
41
'XO:o.+.o:OX\n' + \
 
42
'XO:o...o:OX\n' + \
 
43
'XO:ooooo:OX\n' + \
 
44
'XO:::::::OX\n' + \
 
45
'XOOOOOOOOOX\n' + \
 
46
'XXXXXXXXXXX'
 
47
put_target = '\\.3.5.7.9/\n' + \
 
48
'.........2\n' + \
 
49
'3.........\n' + \
 
50
'.........4\n' + \
 
51
'5...\\/....\n' + \
 
52
'..../\\...6\n' + \
 
53
'7.........\n' + \
 
54
'.........8\n' + \
 
55
'9.........\n' + \
 
56
'/2.4.6.8.\\'
 
57
scroll_target = '\\.3.5.7.9/\n' + \
 
58
'\\.3.5.7.9/\n' + \
 
59
'\\.3.5.7.9/\n' + \
 
60
'\\.3.5.7.9/\n' + \
 
61
'5...\\/....\n' + \
 
62
'..../\\...6\n' + \
 
63
'/2.4.6.8.\\\n' + \
 
64
'/2.4.6.8.\\\n' + \
 
65
'/2.4.6.8.\\\n' + \
 
66
'/2.4.6.8.\\'
 
67
insert_target = 'ZXZZZZZZXZ\n' +\
 
68
'.........2\n' +\
 
69
'3.........\n' +\
 
70
'.........4\n' +\
 
71
'Z5...\\/...\n' +\
 
72
'..../Z\\...\n' +\
 
73
'7.........\n' +\
 
74
'.........8\n' +\
 
75
'9.........\n' +\
 
76
'ZZ/2.4.6ZZ'
 
77
get_region_target = ['......', '.\\/...', './\\...', '......']
 
78
 
 
79
class screenTestCase (PexpectTestCase.PexpectTestCase):
 
80
    def make_screen_with_put (self):
 
81
        s = screen.screen(10,10)
 
82
        s.fill ('.')
 
83
        for r in range (1,s.rows + 1):
 
84
            if r % 2:
 
85
                s.put_abs (r, 1, str(r))
 
86
            else:
 
87
                s.put_abs (r, s.cols, str(r))
 
88
        for c in range (1,s.cols + 1):
 
89
            if c % 2:
 
90
                s.put_abs (1, c, str(c))
 
91
            else:
 
92
                s.put_abs (s.rows, c, str(c))
 
93
        s.put_abs(1,1, '\\')
 
94
        s.put_abs(1,s.cols, '/')
 
95
        s.put_abs(s.rows,1,'/')
 
96
        s.put_abs(s.rows, s.cols, '\\')
 
97
        s.put_abs(5,5,'\\')
 
98
        s.put_abs(5,6,'/')
 
99
        s.put_abs(6,5,'/')
 
100
        s.put_abs(6,6,'\\')
 
101
        return s
 
102
 
 
103
    def test_fill (self):
 
104
        s = screen.screen (10,10)
 
105
        s.fill_region (10,1,1,10,'X')
 
106
        s.fill_region (2,2,9,9,'O')
 
107
        s.fill_region (8,8,3,3,':')
 
108
        s.fill_region (4,7,7,4,'o')
 
109
        s.fill_region (6,5,5,6,'.')
 
110
        assert str(s) == fill1_target
 
111
 
 
112
        s = screen.screen (11,11)
 
113
        s.fill_region (1,1,11,11,'X')
 
114
        s.fill_region (2,2,10,10,'O')
 
115
        s.fill_region (9,9,3,3,':')
 
116
        s.fill_region (4,8,8,4,'o')
 
117
        s.fill_region (7,5,5,7,'.')
 
118
        s.fill_region (6,6,6,6,'+')
 
119
        assert str(s) == fill2_target
 
120
    def test_put (self):
 
121
        s = self.make_screen_with_put()
 
122
        assert str(s) == put_target
 
123
    def test_get_region (self):
 
124
        s = self.make_screen_with_put()
 
125
        r = s.get_region (4,4,7,9)
 
126
        assert r == get_region_target
 
127
 
 
128
    def test_cursor_save (self):
 
129
        s = self.make_screen_with_put()
 
130
        s.cursor_home (5,5)
 
131
        c = s.get()
 
132
        s.cursor_save()
 
133
        s.cursor_home()
 
134
        s.cursor_forward()
 
135
        s.cursor_down()
 
136
        s.cursor_unsave()
 
137
        assert s.cur_r == 5 and s.cur_c == 5
 
138
        assert c == s.get()
 
139
    def test_scroll (self):
 
140
        s = self.make_screen_with_put()
 
141
        s.scroll_screen_rows (1,4)
 
142
        s.scroll_down(); s.scroll_down(); s.scroll_down()
 
143
        s.scroll_down(); s.scroll_down(); s.scroll_down()
 
144
        s.scroll_screen_rows (7,10)
 
145
        s.scroll_up(); s.scroll_up(); s.scroll_up()
 
146
        s.scroll_up(); s.scroll_up(); s.scroll_up()
 
147
        assert str(s) == scroll_target
 
148
    def test_insert (self):
 
149
        s = self.make_screen_with_put()
 
150
        s.insert_abs (10,1,'Z')
 
151
        s.insert_abs (1,1,'Z')
 
152
        s.insert_abs (1,1,'Z')
 
153
        s.insert_abs (1,1,'Z')
 
154
        s.insert_abs (1,1,'Z')
 
155
        s.insert_abs (1,1,'Z')
 
156
        s.insert_abs (10,1,'Z')
 
157
        s.insert_abs (1,1,'Z')
 
158
        s.insert_abs (1,1,'Z')
 
159
        s.insert_abs (5,1,'Z')
 
160
        s.insert_abs (6,6,'Z')
 
161
        s.cursor_home (1,1) # Also test relative insert.
 
162
        s.insert ('Z')
 
163
        s.insert ('Z')
 
164
        s.insert ('Z')
 
165
        s.insert ('Z')
 
166
        s.insert_abs (1,8,'X')
 
167
        s.insert_abs (1,2,'X')
 
168
        s.insert_abs (10,9,'Z')
 
169
        s.insert_abs (10,9,'Z')
 
170
        assert str(s) == insert_target
 
171
 #   def test_write (self):
 
172
 #       s = screen.screen (6,65)
 
173
 #       s.fill('.')
 
174
 #       s.cursor_home()
 
175
 #       for c in write_text:
 
176
 #           s.write (c)
 
177
 #       print str(s)
 
178
 #       assert str(s) == write_target
 
179
 #   def test_tetris (self):
 
180
 #       s = screen.screen (24,80)
 
181
 #       tetris_text = open ('tetris.data').read()
 
182
 #       for c in tetris_text:
 
183
 #           s.write (c)
 
184
 #       assert str(s) == tetris_target
 
185
 
 
186
if __name__ == '__main__':
 
187
    unittest.main()
 
188
 
 
189
suite = unittest.makeSuite(screenTestCase,'test')
 
190
 
 
191