# Tests that work for both bytes and buffer objects.

# See PEP 3137.



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.assert_(self.marshal(b'a').islower())

        self.assertFalse(self.marshal(b'A').islower())

        self.assertFalse(self.marshal(b'\n').islower())

        self.assert_(self.marshal(b'abc').islower())

        self.assertFalse(self.marshal(b'aBc').islower())

        self.assert_(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.assert_(self.marshal(b'A').isupper())

        self.assertFalse(self.marshal(b'\n').isupper())

        self.assert_(self.marshal(b'ABC').isupper())

        self.assertFalse(self.marshal(b'AbC').isupper())

        self.assert_(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.assert_(self.marshal(b'A').istitle())

        self.assertFalse(self.marshal(b'\n').istitle())

        self.assert_(self.marshal(b'A Titlecased Line').istitle())

        self.assert_(self.marshal(b'A\nTitlecased Line').istitle())

        self.assert_(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.assert_(self.marshal(b' ').isspace())

        self.assert_(self.marshal(b'\t').isspace())

        self.assert_(self.marshal(b'\r').isspace())

        self.assert_(self.marshal(b'\n').isspace())

        self.assert_(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.assert_(self.marshal(b'a').isalpha())

        self.assert_(self.marshal(b'A').isalpha())

        self.assertFalse(self.marshal(b'\n').isalpha())

        self.assert_(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.assert_(self.marshal(b'a').isalnum())

        self.assert_(self.marshal(b'A').isalnum())

        self.assertFalse(self.marshal(b'\n').isalnum())

        self.assert_(self.marshal(b'123abc456').isalnum())

        self.assert_(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.assert_(self.marshal(b'0').isdigit())

        self.assert_(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(b'hello', self.marshal(b'HeLLo').lower())

        self.assertEqual(b'hello', self.marshal(b'hello').lower())

        self.assertRaises(TypeError, self.marshal(b'hello').lower, 42)



    def test_upper(self):

        self.assertEqual(b'HELLO', self.marshal(b'HeLLo').upper())

        self.assertEqual(b'HELLO', self.marshal(b'HELLO').upper())

        self.assertRaises(TypeError, self.marshal(b'hello').upper, 42)



    def test_capitalize(self):

        self.assertEqual(b' hello ', self.marshal(b' hello ').capitalize())

        self.assertEqual(b'Hello ', self.marshal(b'Hello ').capitalize())

        self.assertEqual(b'Hello ', self.marshal(b'hello ').capitalize())

        self.assertEqual(b'Aaaa', self.marshal(b'aaaa').capitalize())

        self.assertEqual(b'Aaaa', self.marshal(b'AaAa').capitalize())



        self.assertRaises(TypeError, self.marshal(b'hello').capitalize, 42)



    def test_ljust(self):

        self.assertEqual(b'abc       ', self.marshal(b'abc').ljust(10))

        self.assertEqual(b'abc   ', self.marshal(b'abc').ljust(6))

        self.assertEqual(b'abc', self.marshal(b'abc').ljust(3))

        self.assertEqual(b'abc', self.marshal(b'abc').ljust(2))

        self.assertEqual(b'abc*******', self.marshal(b'abc').ljust(10, '*'))

        self.assertRaises(TypeError, self.marshal(b'abc').ljust)



    def test_rjust(self):

        self.assertEqual(b'       abc', self.marshal(b'abc').rjust(10))

        self.assertEqual(b'   abc', self.marshal(b'abc').rjust(6))

        self.assertEqual(b'abc', self.marshal(b'abc').rjust(3))

        self.assertEqual(b'abc', self.marshal(b'abc').rjust(2))

        self.assertEqual(b'*******abc', self.marshal(b'abc').rjust(10, '*'))

        self.assertRaises(TypeError, self.marshal(b'abc').rjust)



    def test_center(self):

        self.assertEqual(b'   abc    ', self.marshal(b'abc').center(10))

        self.assertEqual(b' abc  ', self.marshal(b'abc').center(6))

        self.assertEqual(b'abc', self.marshal(b'abc').center(3))

        self.assertEqual(b'abc', self.marshal(b'abc').center(2))

        self.assertEqual(b'***abc****', self.marshal(b'abc').center(10, '*'))

        self.assertRaises(TypeError, self.marshal(b'abc').center)



    def test_swapcase(self):

        self.assertEqual(b'hEllO CoMPuTErS',

            self.marshal(b'HeLLo cOmpUteRs').swapcase())



        self.assertRaises(TypeError, self.marshal(b'hello').swapcase, 42)



    def test_zfill(self):

        self.assertEqual(b'123', self.marshal(b'123').zfill(2))

        self.assertEqual(b'123', self.marshal(b'123').zfill(3))

        self.assertEqual(b'0123', self.marshal(b'123').zfill(4))

        self.assertEqual(b'+123', self.marshal(b'+123').zfill(3))

        self.assertEqual(b'+123', self.marshal(b'+123').zfill(4))

        self.assertEqual(b'+0123', self.marshal(b'+123').zfill(5))

        self.assertEqual(b'-123', self.marshal(b'-123').zfill(3))

        self.assertEqual(b'-123', self.marshal(b'-123').zfill(4))

        self.assertEqual(b'-0123', self.marshal(b'-123').zfill(5))

        self.assertEqual(b'000', self.marshal(b'').zfill(3))

        self.assertEqual(b'34', self.marshal(b'34').zfill(1))

        self.assertEqual(b'0034', self.marshal(b'34').zfill(4))



        self.assertRaises(TypeError, self.marshal(b'123').zfill)



    def test_expandtabs(self):

        self.assertEqual(b'abc\rab      def\ng       hi',

                         self.marshal(b'abc\rab\tdef\ng\thi').expandtabs())

        self.assertEqual(b'abc\rab      def\ng       hi',

                         self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(8))

        self.assertEqual(b'abc\rab  def\ng   hi',

                         self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(4))

        self.assertEqual(b'abc\r\nab  def\ng   hi',

                         self.marshal(b'abc\r\nab\tdef\ng\thi').expandtabs(4))

        self.assertEqual(b'abc\rab      def\ng       hi',

                         self.marshal(b'abc\rab\tdef\ng\thi').expandtabs())

        self.assertEqual(b'abc\rab      def\ng       hi',

                         self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(8))

        self.assertEqual(b'abc\r\nab\r\ndef\ng\r\nhi',

            self.marshal(b'abc\r\nab\r\ndef\ng\r\nhi').expandtabs(4))

        self.assertEqual(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.maxint < (1 << 32) and struct.calcsize('P') == 4:

            self.assertRaises(OverflowError,

                              self.marshal(b'\ta\n\tb').expandtabs, sys.maxint)



    def test_title(self):

        self.assertEqual(b' Hello ', self.marshal(b' hello ').title())

        self.assertEqual(b'Hello ', self.marshal(b'hello ').title())

        self.assertEqual(b'Hello ', self.marshal(b'Hello ').title())

        self.assertEqual(b'Format This As Title String',

                         self.marshal(b'fOrMaT thIs aS titLe String').title())

        self.assertEqual(b'Format,This-As*Title;String',

                         self.marshal(b'fOrMaT,thIs-aS*titLe;String').title())

        self.assertEqual(b'Getint', self.marshal(b'getInt').title())

        self.assertRaises(TypeError, self.marshal(b'hello').title, 42)



    def test_splitlines(self):

        self.assertEqual([b'abc', b'def', b'', b'ghi'],

                         self.marshal(b'abc\ndef\n\rghi').splitlines())

        self.assertEqual([b'abc', b'def', b'', b'ghi'],

                         self.marshal(b'abc\ndef\n\r\nghi').splitlines())

        self.assertEqual([b'abc', b'def', b'ghi'],

                         self.marshal(b'abc\ndef\r\nghi').splitlines())

        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'\n', b'abc\n', b'def\r\n', b'ghi\n', b'\r'],

                         self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines(1))



        self.assertRaises(TypeError, self.marshal(b'abc').splitlines, 42, 42)

