Lib/test/buffer_tests.py

   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.assert_(self.marshal(b'a').islower())
  19         self.assertFalse(self.marshal(b'A').islower())
  20         self.assertFalse(self.marshal(b'\n').islower())
  21         self.assert_(self.marshal(b'abc').islower())
  22         self.assertFalse(self.marshal(b'aBc').islower())
  23         self.assert_(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.assert_(self.marshal(b'A').isupper())
  30         self.assertFalse(self.marshal(b'\n').isupper())
  31         self.assert_(self.marshal(b'ABC').isupper())
  32         self.assertFalse(self.marshal(b'AbC').isupper())
  33         self.assert_(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.assert_(self.marshal(b'A').istitle())
  40         self.assertFalse(self.marshal(b'\n').istitle())
  41         self.assert_(self.marshal(b'A Titlecased Line').istitle())
  42         self.assert_(self.marshal(b'A\nTitlecased Line').istitle())
  43         self.assert_(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.assert_(self.marshal(b' ').isspace())
  54         self.assert_(self.marshal(b'\t').isspace())
  55         self.assert_(self.marshal(b'\r').isspace())
  56         self.assert_(self.marshal(b'\n').isspace())
  57         self.assert_(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.assert_(self.marshal(b'a').isalpha())
  64         self.assert_(self.marshal(b'A').isalpha())
  65         self.assertFalse(self.marshal(b'\n').isalpha())
  66         self.assert_(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.assert_(self.marshal(b'a').isalnum())
  74         self.assert_(self.marshal(b'A').isalnum())
  75         self.assertFalse(self.marshal(b'\n').isalnum())
  76         self.assert_(self.marshal(b'123abc456').isalnum())
  77         self.assert_(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.assert_(self.marshal(b'0').isdigit())
  86         self.assert_(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)