Lib.test.test_StringIO_jy.py Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jython-standalone Show documentation

Jython is an implementation of the high-level, dynamic, object-oriented language Python written in 100% Pure Java, and seamlessly integrated with the Java platform. It thus allows you to run Python on any Java platform.

The newest version!

import unittest
import cStringIO
from test import test_support

class TestUnicodeInput(unittest.TestCase):
    def test_differences_handling_unicode(self):
        # Test for the "feature" described on #1089.
        #
        # Basically, StringIO returns unicode objects if you feed it unicode,
        # but cStringIO don't. This should change in future versions of
        # CPython and Jython.
        self.assertEqual(u'foo', cStringIO.StringIO(u'foo').read())
        self.assertEqual('foo', cStringIO.StringIO(u'foo').read())

    def test_unicode_encode_error(self):
        # See issue #2527
        uc = u'\u4e2d\u6587'
        self.assertRaises(UnicodeEncodeError, lambda: cStringIO.StringIO(uc))


class TestWrite(unittest.TestCase):
    def test_write_seek_write(self):
        f = cStringIO.StringIO()
        f.write('hello')
        f.seek(2)
        f.write('hi')
        self.assertEquals(f.getvalue(), 'hehio')

    #XXX: this should get pushed to CPython's test_StringIO
    def test_write_past_end(self):
        f = cStringIO.StringIO()
        f.write("abcdef")
        f.seek(10)
        f.write("uvwxyz")
        self.assertEqual(f.getvalue(), 'abcdef\x00\x00\x00\x00uvwxyz')

    def test_write_seek_back_then_write(self):
        # http://bugs.jython.org/issue2324
        s = "abcdef"
        for i in xrange(len(s)):
            f = cStringIO.StringIO()
            f.write(s)
            f.seek(i)
            f.write("x" * 47)
            self.assertEqual(f.getvalue(), s[:i] + ("x" * 47))


class TestGetValueAfterClose(unittest.TestCase):

    # This test, or something like it, should be really be pushed upstream
    def test_getvalue_after_close(self):
        f = cStringIO.StringIO('hello')
        f.getvalue()
        f.close()
        try:
            f.getvalue()
        except ValueError:
            pass
        else:
            self.fail("cStringIO.StringIO: getvalue() after close() should have raised ValueError")

def test_main():
    test_support.run_unittest(TestUnicodeInput)
    test_support.run_unittest(TestWrite)
    test_support.run_unittest(TestGetValueAfterClose)

if __name__ == '__main__':
    test_main()