1 # Tests that work for both bytes and buffer objects. 2 # See PEP 3137. 3 4 import struct 5 import sys 6 7 class MixinBytesBufferCommonTests(object): 8 """Tests that work for both bytes and buffer objects. 9 See PEP 3137. 10 """ 11 12 def marshal(self, x): 13 """Convert x into the appropriate type for these tests.""" 14 raise RuntimeError('test class must provide a marshal method') 15 16 def test_islower(self): 17 self.assertFalse(self.marshal(b'').islower()) 18 self.assertTrue(self.marshal(b'a').islower()) 19 self.assertFalse(self.marshal(b'A').islower()) 20 self.assertFalse(self.marshal(b'\n').islower()) 21 self.assertTrue(self.marshal(b'abc').islower()) 22 self.assertFalse(self.marshal(b'aBc').islower()) 23 self.assertTrue(self.marshal(b'abc\n').islower()) 24 self.assertRaises(TypeError, self.marshal(b'abc').islower, 42) 25 26 def test_isupper(self): 27 self.assertFalse(self.marshal(b'').isupper()) 28 self.assertFalse(self.marshal(b'a').isupper()) 29 self.assertTrue(self.marshal(b'A').isupper()) 30 self.assertFalse(self.marshal(b'\n').isupper()) 31 self.assertTrue(self.marshal(b'ABC').isupper()) 32 self.assertFalse(self.marshal(b'AbC').isupper()) 33 self.assertTrue(self.marshal(b'ABC\n').isupper()) 34 self.assertRaises(TypeError, self.marshal(b'abc').isupper, 42) 35 36 def test_istitle(self): 37 self.assertFalse(self.marshal(b'').istitle()) 38 self.assertFalse(self.marshal(b'a').istitle()) 39 self.assertTrue(self.marshal(b'A').istitle()) 40 self.assertFalse(self.marshal(b'\n').istitle()) 41 self.assertTrue(self.marshal(b'A Titlecased Line').istitle()) 42 self.assertTrue(self.marshal(b'A\nTitlecased Line').istitle()) 43 self.assertTrue(self.marshal(b'A Titlecased, Line').istitle()) 44 self.assertFalse(self.marshal(b'Not a capitalized String').istitle()) 45 self.assertFalse(self.marshal(b'Not\ta Titlecase String').istitle()) 46 self.assertFalse(self.marshal(b'Not--a Titlecase String').istitle()) 47 self.assertFalse(self.marshal(b'NOT').istitle()) 48 self.assertRaises(TypeError, self.marshal(b'abc').istitle, 42) 49 50 def test_isspace(self): 51 self.assertFalse(self.marshal(b'').isspace()) 52 self.assertFalse(self.marshal(b'a').isspace()) 53 self.assertTrue(self.marshal(b' ').isspace()) 54 self.assertTrue(self.marshal(b'\t').isspace()) 55 self.assertTrue(self.marshal(b'\r').isspace()) 56 self.assertTrue(self.marshal(b'\n').isspace()) 57 self.assertTrue(self.marshal(b' \t\r\n').isspace()) 58 self.assertFalse(self.marshal(b' \t\r\na').isspace()) 59 self.assertRaises(TypeError, self.marshal(b'abc').isspace, 42) 60 61 def test_isalpha(self): 62 self.assertFalse(self.marshal(b'').isalpha()) 63 self.assertTrue(self.marshal(b'a').isalpha()) 64 self.assertTrue(self.marshal(b'A').isalpha()) 65 self.assertFalse(self.marshal(b'\n').isalpha()) 66 self.assertTrue(self.marshal(b'abc').isalpha()) 67 self.assertFalse(self.marshal(b'aBc123').isalpha()) 68 self.assertFalse(self.marshal(b'abc\n').isalpha()) 69 self.assertRaises(TypeError, self.marshal(b'abc').isalpha, 42) 70 71 def test_isalnum(self): 72 self.assertFalse(self.marshal(b'').isalnum()) 73 self.assertTrue(self.marshal(b'a').isalnum()) 74 self.assertTrue(self.marshal(b'A').isalnum()) 75 self.assertFalse(self.marshal(b'\n').isalnum()) 76 self.assertTrue(self.marshal(b'123abc456').isalnum()) 77 self.assertTrue(self.marshal(b'a1b3c').isalnum()) 78 self.assertFalse(self.marshal(b'aBc000 ').isalnum()) 79 self.assertFalse(self.marshal(b'abc\n').isalnum()) 80 self.assertRaises(TypeError, self.marshal(b'abc').isalnum, 42) 81 82 def test_isdigit(self): 83 self.assertFalse(self.marshal(b'').isdigit()) 84 self.assertFalse(self.marshal(b'a').isdigit()) 85 self.assertTrue(self.marshal(b'0').isdigit()) 86 self.assertTrue(self.marshal(b'0123456789').isdigit()) 87 self.assertFalse(self.marshal(b'0123456789a').isdigit()) 88 89 self.assertRaises(TypeError, self.marshal(b'abc').isdigit, 42) 90 91 def test_lower(self): 92 self.assertEqual(b'hello', self.marshal(b'HeLLo').lower()) 93 self.assertEqual(b'hello', self.marshal(b'hello').lower()) 94 self.assertRaises(TypeError, self.marshal(b'hello').lower, 42) 95 96 def test_upper(self): 97 self.assertEqual(b'HELLO', self.marshal(b'HeLLo').upper()) 98 self.assertEqual(b'HELLO', self.marshal(b'HELLO').upper()) 99 self.assertRaises(TypeError, self.marshal(b'hello').upper, 42) 100 101 def test_capitalize(self): 102 self.assertEqual(b' hello ', self.marshal(b' hello ').capitalize()) 103 self.assertEqual(b'Hello ', self.marshal(b'Hello ').capitalize()) 104 self.assertEqual(b'Hello ', self.marshal(b'hello ').capitalize()) 105 self.assertEqual(b'Aaaa', self.marshal(b'aaaa').capitalize()) 106 self.assertEqual(b'Aaaa', self.marshal(b'AaAa').capitalize()) 107 108 self.assertRaises(TypeError, self.marshal(b'hello').capitalize, 42) 109 110 def test_ljust(self): 111 self.assertEqual(b'abc ', self.marshal(b'abc').ljust(10)) 112 self.assertEqual(b'abc ', self.marshal(b'abc').ljust(6)) 113 self.assertEqual(b'abc', self.marshal(b'abc').ljust(3)) 114 self.assertEqual(b'abc', self.marshal(b'abc').ljust(2)) 115 self.assertEqual(b'abc*******', self.marshal(b'abc').ljust(10, '*')) 116 self.assertRaises(TypeError, self.marshal(b'abc').ljust) 117 118 def test_rjust(self): 119 self.assertEqual(b' abc', self.marshal(b'abc').rjust(10)) 120 self.assertEqual(b' abc', self.marshal(b'abc').rjust(6)) 121 self.assertEqual(b'abc', self.marshal(b'abc').rjust(3)) 122 self.assertEqual(b'abc', self.marshal(b'abc').rjust(2)) 123 self.assertEqual(b'*******abc', self.marshal(b'abc').rjust(10, '*')) 124 self.assertRaises(TypeError, self.marshal(b'abc').rjust) 125 126 def test_center(self): 127 self.assertEqual(b' abc ', self.marshal(b'abc').center(10)) 128 self.assertEqual(b' abc ', self.marshal(b'abc').center(6)) 129 self.assertEqual(b'abc', self.marshal(b'abc').center(3)) 130 self.assertEqual(b'abc', self.marshal(b'abc').center(2)) 131 self.assertEqual(b'***abc****', self.marshal(b'abc').center(10, '*')) 132 self.assertRaises(TypeError, self.marshal(b'abc').center) 133 134 def test_swapcase(self): 135 self.assertEqual(b'hEllO CoMPuTErS', 136 self.marshal(b'HeLLo cOmpUteRs').swapcase()) 137 138 self.assertRaises(TypeError, self.marshal(b'hello').swapcase, 42) 139 140 def test_zfill(self): 141 self.assertEqual(b'123', self.marshal(b'123').zfill(2)) 142 self.assertEqual(b'123', self.marshal(b'123').zfill(3)) 143 self.assertEqual(b'0123', self.marshal(b'123').zfill(4)) 144 self.assertEqual(b'+123', self.marshal(b'+123').zfill(3)) 145 self.assertEqual(b'+123', self.marshal(b'+123').zfill(4)) 146 self.assertEqual(b'+0123', self.marshal(b'+123').zfill(5)) 147 self.assertEqual(b'-123', self.marshal(b'-123').zfill(3)) 148 self.assertEqual(b'-123', self.marshal(b'-123').zfill(4)) 149 self.assertEqual(b'-0123', self.marshal(b'-123').zfill(5)) 150 self.assertEqual(b'000', self.marshal(b'').zfill(3)) 151 self.assertEqual(b'34', self.marshal(b'34').zfill(1)) 152 self.assertEqual(b'0034', self.marshal(b'34').zfill(4)) 153 154 self.assertRaises(TypeError, self.marshal(b'123').zfill) 155 156 def test_expandtabs(self): 157 self.assertEqual(b'abc\rab def\ng hi', 158 self.marshal(b'abc\rab\tdef\ng\thi').expandtabs()) 159 self.assertEqual(b'abc\rab def\ng hi', 160 self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(8)) 161 self.assertEqual(b'abc\rab def\ng hi', 162 self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(4)) 163 self.assertEqual(b'abc\r\nab def\ng hi', 164 self.marshal(b'abc\r\nab\tdef\ng\thi').expandtabs(4)) 165 self.assertEqual(b'abc\rab def\ng hi', 166 self.marshal(b'abc\rab\tdef\ng\thi').expandtabs()) 167 self.assertEqual(b'abc\rab def\ng hi', 168 self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(8)) 169 self.assertEqual(b'abc\r\nab\r\ndef\ng\r\nhi', 170 self.marshal(b'abc\r\nab\r\ndef\ng\r\nhi').expandtabs(4)) 171 self.assertEqual(b' a\n b', self.marshal(b' \ta\n\tb').expandtabs(1)) 172 173 self.assertRaises(TypeError, self.marshal(b'hello').expandtabs, 42, 42) 174 # This test is only valid when sizeof(int) == sizeof(void*) == 4. 175 if sys.maxint < (1 << 32) and struct.calcsize('P') == 4: 176 self.assertRaises(OverflowError, 177 self.marshal(b'\ta\n\tb').expandtabs, sys.maxint) 178 179 def test_title(self): 180 self.assertEqual(b' Hello ', self.marshal(b' hello ').title()) 181 self.assertEqual(b'Hello ', self.marshal(b'hello ').title()) 182 self.assertEqual(b'Hello ', self.marshal(b'Hello ').title()) 183 self.assertEqual(b'Format This As Title String', 184 self.marshal(b'fOrMaT thIs aS titLe String').title()) 185 self.assertEqual(b'Format,This-As*Title;String', 186 self.marshal(b'fOrMaT,thIs-aS*titLe;String').title()) 187 self.assertEqual(b'Getint', self.marshal(b'getInt').title()) 188 self.assertRaises(TypeError, self.marshal(b'hello').title, 42) 189 190 def test_splitlines(self): 191 self.assertEqual([b'abc', b'def', b'', b'ghi'], 192 self.marshal(b'abc\ndef\n\rghi').splitlines()) 193 self.assertEqual([b'abc', b'def', b'', b'ghi'], 194 self.marshal(b'abc\ndef\n\r\nghi').splitlines()) 195 self.assertEqual([b'abc', b'def', b'ghi'], 196 self.marshal(b'abc\ndef\r\nghi').splitlines()) 197 self.assertEqual([b'abc', b'def', b'ghi'], 198 self.marshal(b'abc\ndef\r\nghi\n').splitlines()) 199 self.assertEqual([b'abc', b'def', b'ghi', b''], 200 self.marshal(b'abc\ndef\r\nghi\n\r').splitlines()) 201 self.assertEqual([b'', b'abc', b'def', b'ghi', b''], 202 self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines()) 203 self.assertEqual([b'\n', b'abc\n', b'def\r\n', b'ghi\n', b'\r'], 204 self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines(1)) 205 206 self.assertRaises(TypeError, self.marshal(b'abc').splitlines, 42, 42) 207
