252 lines
13 KiB
Python
252 lines
13 KiB
Python
# Tests that work for both bytes and buffer objects.
|
|
# See PEP 3137.
|
|
|
|
from __future__ import (absolute_import, division,
|
|
print_function, unicode_literals)
|
|
from future.builtins import *
|
|
from future.tests.base import unittest, expectedFailurePY26
|
|
|
|
import struct
|
|
import sys
|
|
|
|
|
|
class MixinBytesBufferCommonTests(object):
|
|
"""Tests that work for both bytes and buffer objects.
|
|
See PEP 3137.
|
|
"""
|
|
|
|
def marshal(self, x):
|
|
"""Convert x into the appropriate type for these tests."""
|
|
raise RuntimeError('test class must provide a marshal method')
|
|
|
|
def test_islower(self):
|
|
self.assertFalse(self.marshal(b'').islower())
|
|
self.assertTrue(self.marshal(b'a').islower())
|
|
self.assertFalse(self.marshal(b'A').islower())
|
|
self.assertFalse(self.marshal(b'\n').islower())
|
|
self.assertTrue(self.marshal(b'abc').islower())
|
|
self.assertFalse(self.marshal(b'aBc').islower())
|
|
self.assertTrue(self.marshal(b'abc\n').islower())
|
|
self.assertRaises(TypeError, self.marshal(b'abc').islower, 42)
|
|
|
|
def test_isupper(self):
|
|
self.assertFalse(self.marshal(b'').isupper())
|
|
self.assertFalse(self.marshal(b'a').isupper())
|
|
self.assertTrue(self.marshal(b'A').isupper())
|
|
self.assertFalse(self.marshal(b'\n').isupper())
|
|
self.assertTrue(self.marshal(b'ABC').isupper())
|
|
self.assertFalse(self.marshal(b'AbC').isupper())
|
|
self.assertTrue(self.marshal(b'ABC\n').isupper())
|
|
self.assertRaises(TypeError, self.marshal(b'abc').isupper, 42)
|
|
|
|
def test_istitle(self):
|
|
self.assertFalse(self.marshal(b'').istitle())
|
|
self.assertFalse(self.marshal(b'a').istitle())
|
|
self.assertTrue(self.marshal(b'A').istitle())
|
|
self.assertFalse(self.marshal(b'\n').istitle())
|
|
self.assertTrue(self.marshal(b'A Titlecased Line').istitle())
|
|
self.assertTrue(self.marshal(b'A\nTitlecased Line').istitle())
|
|
self.assertTrue(self.marshal(b'A Titlecased, Line').istitle())
|
|
self.assertFalse(self.marshal(b'Not a capitalized String').istitle())
|
|
self.assertFalse(self.marshal(b'Not\ta Titlecase String').istitle())
|
|
self.assertFalse(self.marshal(b'Not--a Titlecase String').istitle())
|
|
self.assertFalse(self.marshal(b'NOT').istitle())
|
|
self.assertRaises(TypeError, self.marshal(b'abc').istitle, 42)
|
|
|
|
def test_isspace(self):
|
|
self.assertFalse(self.marshal(b'').isspace())
|
|
self.assertFalse(self.marshal(b'a').isspace())
|
|
self.assertTrue(self.marshal(b' ').isspace())
|
|
self.assertTrue(self.marshal(b'\t').isspace())
|
|
self.assertTrue(self.marshal(b'\r').isspace())
|
|
self.assertTrue(self.marshal(b'\n').isspace())
|
|
self.assertTrue(self.marshal(b' \t\r\n').isspace())
|
|
self.assertFalse(self.marshal(b' \t\r\na').isspace())
|
|
self.assertRaises(TypeError, self.marshal(b'abc').isspace, 42)
|
|
|
|
def test_isalpha(self):
|
|
self.assertFalse(self.marshal(b'').isalpha())
|
|
self.assertTrue(self.marshal(b'a').isalpha())
|
|
self.assertTrue(self.marshal(b'A').isalpha())
|
|
self.assertFalse(self.marshal(b'\n').isalpha())
|
|
self.assertTrue(self.marshal(b'abc').isalpha())
|
|
self.assertFalse(self.marshal(b'aBc123').isalpha())
|
|
self.assertFalse(self.marshal(b'abc\n').isalpha())
|
|
self.assertRaises(TypeError, self.marshal(b'abc').isalpha, 42)
|
|
|
|
def test_isalnum(self):
|
|
self.assertFalse(self.marshal(b'').isalnum())
|
|
self.assertTrue(self.marshal(b'a').isalnum())
|
|
self.assertTrue(self.marshal(b'A').isalnum())
|
|
self.assertFalse(self.marshal(b'\n').isalnum())
|
|
self.assertTrue(self.marshal(b'123abc456').isalnum())
|
|
self.assertTrue(self.marshal(b'a1b3c').isalnum())
|
|
self.assertFalse(self.marshal(b'aBc000 ').isalnum())
|
|
self.assertFalse(self.marshal(b'abc\n').isalnum())
|
|
self.assertRaises(TypeError, self.marshal(b'abc').isalnum, 42)
|
|
|
|
def test_isdigit(self):
|
|
self.assertFalse(self.marshal(b'').isdigit())
|
|
self.assertFalse(self.marshal(b'a').isdigit())
|
|
self.assertTrue(self.marshal(b'0').isdigit())
|
|
self.assertTrue(self.marshal(b'0123456789').isdigit())
|
|
self.assertFalse(self.marshal(b'0123456789a').isdigit())
|
|
|
|
self.assertRaises(TypeError, self.marshal(b'abc').isdigit, 42)
|
|
|
|
def test_lower(self):
|
|
self.assertEqual(bytes(b'hello'), self.marshal(b'HeLLo').lower())
|
|
self.assertEqual(bytes(b'hello'), self.marshal(b'hello').lower())
|
|
self.assertRaises(TypeError, self.marshal(b'hello').lower, 42)
|
|
|
|
def test_upper(self):
|
|
self.assertEqual(bytes(b'HELLO'), self.marshal(b'HeLLo').upper())
|
|
self.assertEqual(bytes(b'HELLO'), self.marshal(b'HELLO').upper())
|
|
self.assertRaises(TypeError, self.marshal(b'hello').upper, 42)
|
|
|
|
def test_capitalize(self):
|
|
self.assertEqual(bytes(b' hello '), self.marshal(b' hello ').capitalize())
|
|
self.assertEqual(bytes(b'Hello '), self.marshal(b'Hello ').capitalize())
|
|
self.assertEqual(bytes(b'Hello '), self.marshal(b'hello ').capitalize())
|
|
self.assertEqual(bytes(b'Aaaa'), self.marshal(b'aaaa').capitalize())
|
|
self.assertEqual(bytes(b'Aaaa'), self.marshal(b'AaAa').capitalize())
|
|
|
|
self.assertRaises(TypeError, self.marshal(b'hello').capitalize, 42)
|
|
|
|
def test_ljust(self):
|
|
self.assertEqual(bytes(b'abc '), self.marshal(b'abc').ljust(10))
|
|
self.assertEqual(bytes(b'abc '), self.marshal(b'abc').ljust(6))
|
|
self.assertEqual(bytes(b'abc'), self.marshal(b'abc').ljust(3))
|
|
self.assertEqual(bytes(b'abc'), self.marshal(b'abc').ljust(2))
|
|
self.assertEqual(bytes(b'abc*******'), self.marshal(b'abc').ljust(10, b'*'))
|
|
self.assertRaises(TypeError, self.marshal(b'abc').ljust)
|
|
|
|
def test_rjust(self):
|
|
self.assertEqual(bytes(b' abc'), self.marshal(b'abc').rjust(10))
|
|
self.assertEqual(bytes(b' abc'), self.marshal(b'abc').rjust(6))
|
|
self.assertEqual(bytes(b'abc'), self.marshal(b'abc').rjust(3))
|
|
self.assertEqual(bytes(b'abc'), self.marshal(b'abc').rjust(2))
|
|
self.assertEqual(bytes(b'*******abc'), self.marshal(b'abc').rjust(10, b'*'))
|
|
self.assertRaises(TypeError, self.marshal(b'abc').rjust)
|
|
|
|
def test_center(self):
|
|
self.assertEqual(bytes(b' abc '), self.marshal(b'abc').center(10))
|
|
self.assertEqual(bytes(b' abc '), self.marshal(b'abc').center(6))
|
|
self.assertEqual(bytes(b'abc'), self.marshal(b'abc').center(3))
|
|
self.assertEqual(bytes(b'abc'), self.marshal(b'abc').center(2))
|
|
self.assertEqual(bytes(b'***abc****'), self.marshal(b'abc').center(10, b'*'))
|
|
self.assertRaises(TypeError, self.marshal(b'abc').center)
|
|
|
|
def test_swapcase(self):
|
|
self.assertEqual(bytes(b'hEllO CoMPuTErS'),
|
|
self.marshal(bytes(b'HeLLo cOmpUteRs')).swapcase())
|
|
|
|
self.assertRaises(TypeError, self.marshal(b'hello').swapcase, 42)
|
|
|
|
def test_zfill(self):
|
|
self.assertEqual(bytes(b'123'), self.marshal(b'123').zfill(2))
|
|
self.assertEqual(bytes(b'123'), self.marshal(b'123').zfill(3))
|
|
self.assertEqual(bytes(b'0123'), self.marshal(b'123').zfill(4))
|
|
self.assertEqual(bytes(b'+123'), self.marshal(b'+123').zfill(3))
|
|
self.assertEqual(bytes(b'+123'), self.marshal(b'+123').zfill(4))
|
|
self.assertEqual(bytes(b'+0123'), self.marshal(b'+123').zfill(5))
|
|
self.assertEqual(bytes(b'-123'), self.marshal(b'-123').zfill(3))
|
|
self.assertEqual(bytes(b'-123'), self.marshal(b'-123').zfill(4))
|
|
self.assertEqual(bytes(b'-0123'), self.marshal(b'-123').zfill(5))
|
|
self.assertEqual(bytes(b'000'), self.marshal(b'').zfill(3))
|
|
self.assertEqual(bytes(b'34'), self.marshal(b'34').zfill(1))
|
|
self.assertEqual(bytes(b'0034'), self.marshal(b'34').zfill(4))
|
|
|
|
self.assertRaises(TypeError, self.marshal(b'123').zfill)
|
|
|
|
def test_expandtabs(self):
|
|
self.assertEqual(bytes(b'abc\rab def\ng hi'),
|
|
self.marshal(b'abc\rab\tdef\ng\thi').expandtabs())
|
|
self.assertEqual(bytes(b'abc\rab def\ng hi'),
|
|
self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(8))
|
|
self.assertEqual(bytes(b'abc\rab def\ng hi'),
|
|
self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(4))
|
|
self.assertEqual(bytes(b'abc\r\nab def\ng hi'),
|
|
self.marshal(b'abc\r\nab\tdef\ng\thi').expandtabs(4))
|
|
self.assertEqual(bytes(b'abc\rab def\ng hi'),
|
|
self.marshal(b'abc\rab\tdef\ng\thi').expandtabs())
|
|
self.assertEqual(bytes(b'abc\rab def\ng hi'),
|
|
self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(8))
|
|
self.assertEqual(bytes(b'abc\r\nab\r\ndef\ng\r\nhi'),
|
|
self.marshal(b'abc\r\nab\r\ndef\ng\r\nhi').expandtabs(4))
|
|
self.assertEqual(bytes(b' a\n b'), self.marshal(b' \ta\n\tb').expandtabs(1))
|
|
|
|
self.assertRaises(TypeError, self.marshal(b'hello').expandtabs, 42, 42)
|
|
# This test is only valid when sizeof(int) == sizeof(void*) == 4.
|
|
if sys.maxsize < (1 << 32) and struct.calcsize('P') == 4:
|
|
self.assertRaises(OverflowError,
|
|
self.marshal(b'\ta\n\tb').expandtabs, sys.maxsize)
|
|
|
|
def test_title(self):
|
|
self.assertEqual(bytes(b' Hello '), self.marshal(b' hello ').title())
|
|
self.assertEqual(bytes(b'Hello '), self.marshal(b'hello ').title())
|
|
self.assertEqual(bytes(b'Hello '), self.marshal(b'Hello ').title())
|
|
self.assertEqual(bytes(b'Format This As Title String'),
|
|
self.marshal(b'fOrMaT thIs aS titLe String').title())
|
|
self.assertEqual(bytes(b'Format,This-As*Title;String'),
|
|
self.marshal(b'fOrMaT,thIs-aS*titLe;String').title())
|
|
self.assertEqual(bytes(b'Getint'), self.marshal(b'getInt').title())
|
|
self.assertRaises(TypeError, self.marshal(b'hello').title, 42)
|
|
|
|
def test_splitlines(self):
|
|
self.assertEqual([bytes(b'abc'), bytes(b'def'), bytes(b''), bytes(b'ghi')],
|
|
self.marshal(b'abc\ndef\n\rghi').splitlines())
|
|
self.assertEqual([bytes(b'abc'), bytes(b'def'), bytes(b''), bytes(b'ghi')],
|
|
self.marshal(b'abc\ndef\n\r\nghi').splitlines())
|
|
self.assertEqual([bytes(b'abc'), bytes(b'def'), bytes(b'ghi')],
|
|
self.marshal(b'abc\ndef\r\nghi').splitlines())
|
|
# TODO: add bytes calls around these too ...
|
|
self.assertEqual([b'abc', b'def', b'ghi'],
|
|
self.marshal(b'abc\ndef\r\nghi\n').splitlines())
|
|
self.assertEqual([b'abc', b'def', b'ghi', b''],
|
|
self.marshal(b'abc\ndef\r\nghi\n\r').splitlines())
|
|
self.assertEqual([b'', b'abc', b'def', b'ghi', b''],
|
|
self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines())
|
|
self.assertEqual([b'', b'abc', b'def', b'ghi', b''],
|
|
self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines(False))
|
|
self.assertEqual([b'\n', b'abc\n', b'def\r\n', b'ghi\n', b'\r'],
|
|
self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines(True))
|
|
self.assertEqual([b'', b'abc', b'def', b'ghi', b''],
|
|
self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines(False))
|
|
self.assertEqual([b'\n', b'abc\n', b'def\r\n', b'ghi\n', b'\r'],
|
|
self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines(True))
|
|
|
|
self.assertRaises(TypeError, self.marshal(b'abc').splitlines, 42, 42)
|
|
|
|
|
|
# From Python-3.3.5/Lib/test/test_bytes.py:
|
|
|
|
class BytearrayPEP3137Test(unittest.TestCase,
|
|
MixinBytesBufferCommonTests):
|
|
def marshal(self, x):
|
|
return bytearray(bytes(x))
|
|
|
|
@expectedFailurePY26
|
|
def test_returns_new_copy(self):
|
|
val = self.marshal(b'1234')
|
|
# On immutable types these MAY return a reference to themselves
|
|
# but on mutable types like bytearray they MUST return a new copy.
|
|
for methname in ('zfill', 'rjust', 'ljust', 'center'):
|
|
method = getattr(val, methname)
|
|
newval = method(3)
|
|
self.assertEqual(val, newval)
|
|
self.assertTrue(val is not newval,
|
|
methname+' returned self on a mutable object')
|
|
for expr in ('val.split()[0]', 'val.rsplit()[0]',
|
|
'val.partition(b".")[0]', 'val.rpartition(b".")[2]',
|
|
'val.splitlines()[0]', 'val.replace(b"", b"")'):
|
|
newval = eval(expr)
|
|
self.assertEqual(val, newval)
|
|
self.assertTrue(val is not newval,
|
|
expr+' returned val on a mutable object')
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|