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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
import sys
if __name__ == '__main__':
import os
pkg_dir = os.path.split(os.path.abspath(__file__))[0]
parent_dir, pkg_name = os.path.split(pkg_dir)
is_pygame_pkg = (pkg_name == 'tests' and
os.path.split(parent_dir)[1] == 'pygame')
if not is_pygame_pkg:
sys.path.insert(0, parent_dir)
else:
is_pygame_pkg = __name__.startswith('pygame.tests.')
import unittest
from pygame import compat
encode_file_path = sys.modules['pygame.rwobject'].encode_file_path
class CompatModuleTest(unittest.TestCase):
def test_as_unicode(self):
r = r'Bo\u00F6tes'
ords = [ord('B'), ord('o'), 0xF6, ord('t'), ord('e'), ord('s')]
self.failUnlessEqual(len(r), 11)
u = compat.as_unicode(r)
self.failUnless(isinstance(u, compat.unicode_))
self.failUnlessEqual([ord(c) for c in u], ords)
def test_as_bytes(self):
ords = [0, 1, 0x7F, 0x80, 0xC3, 0x20, 0xC3, 0xB6, 0xFF]
s = ''.join([chr(i) for i in ords])
self.failUnlessEqual(len(s), len(ords))
b = compat.as_bytes(s)
self.failUnless(isinstance(b, compat.bytes_))
self.failUnlessEqual([compat.ord_(i) for i in b], ords)
def test_ord_(self):
self.failUnless(isinstance(compat.ord_(compat.bytes_(1)[0]), int))
def test_bytes_(self):
self.failIf(compat.bytes_ is compat.unicode_)
self.failUnless(hasattr(compat.bytes_, 'capitalize'))
self.failIf(hasattr(compat.bytes_, 'isdecimal'))
def test_unicode_(self):
self.failUnless(hasattr(compat.unicode_(), 'isdecimal'))
def test_long_(self):
self.failUnless(isinstance(int('99999999999999999999'), compat.long_))
def test_geterror(self):
msg = 'Success'
try:
raise TypeError(msg)
except TypeError:
e = compat.geterror()
self.failUnless(isinstance(e, TypeError))
self.failUnlessEqual(str(e), msg)
def test_xrange_(self):
self.failIf(isinstance(compat.xrange_(2), list))
def test_unichr_(self):
ordval = 86
c = compat.unichr_(ordval)
self.failUnless(isinstance(c, compat.unicode_))
self.failUnlessEqual(ord(c), ordval)
def test_get_BytesIO(self):
BytesIO = compat.get_BytesIO()
b1 = compat.as_bytes("\x00\xffabc")
b2 = BytesIO(b1).read()
self.failUnless(isinstance(b2, compat.bytes_))
self.failUnlessEqual(b2, b1)
def test_get_StringIO(self):
StringIO = compat.get_StringIO()
b1 = "abcde"
b2 = StringIO(b1).read()
self.failUnless(isinstance(b2, str))
self.failUnlessEqual(b2, b1)
def test_raw_input_(self):
StringIO = compat.get_StringIO()
msg = 'success'
tmp = sys.stdin
sys.stdin = StringIO(msg + '\n')
try:
s = compat.raw_input_()
self.failUnlessEqual(s, msg)
finally:
sys.stdin = tmp
def test_filesystem_encode(self):
upath = compat.as_unicode(r"ab\u212Acd")
self.assertEqual(compat.filesystem_encode(upath),
encode_file_path(upath))
if __name__ == '__main__':
unittest.main()
|
