docutils.Lib.test.test_long_int_args.py Maven / Gradle / Ivy
"""Ensure longs passed to int arguments are handled correctly
Made for Jython.
"""
import array
import cStringIO
import tempfile
from test import test_support
import unittest
import StringIO
class LongIntArgsTestCase(unittest.TestCase):
def test_array(self):
a = array.array('c', 'jython')
assert a.pop(0L) == 'j'
def test_file(self):
test_file = tempfile.TemporaryFile()
try:
self._test_file(test_file)
finally:
test_file.close()
def test_StringIO(self):
self._test_file(StringIO.StringIO())
def test_cStringIO(self):
self._test_file(cStringIO.StringIO())
def test_str(self):
self._test_basestring(str)
def test_unicode(self):
self._test_basestring(unicode)
def _test_basestring(self, class_):
s = class_('hello from jython')
l = long(len(s))
assert s.count('o', 0L, l) == 3
assert s.endswith('n', 0L, l) == True
assert s.expandtabs(1L) == s
assert s.find('h', 0L, l) == 0
assert s.index('h', 0L, l) == 0
assert s.rfind('h', 0L, l) == 14
assert s.rindex('h', 0L, l) == 14
assert s.split(' ', 1L) == ['hello', 'from jython']
assert s.startswith('jython', 11L)
def _test_file(self, test_file):
test_file.write('jython')
test_file.seek(0L)
assert test_file.read(1L) == 'j'
assert test_file.readline(2L) == 'yt'
assert test_file.readlines(3L) == ['hon']
def test_main():
test_support.run_unittest(LongIntArgsTestCase)
if __name__ == '__main__':
test_main()