def setUp(self): names = ["alpha{0}.txt".format(x) for x in range(1,7)] names = [os.path.join("data", "txt", name) for name in names] self.names = names self.sris = SplitRawIStream(names)
def setUp(self): names = ["alpha{0}.txt".format(x) for x in range(1, 7)] names = [os.path.join("data", "txt", name) for name in names] self.names = names self.sris = SplitRawIStream(names)
class SplitRawIStreamTestCase(TestCase): def setUp(self): names = ["alpha{0}.txt".format(x) for x in range(1,7)] names = [os.path.join("data", "txt", name) for name in names] self.names = names self.sris = SplitRawIStream(names) # end def setUp def test__init__(self): ae = self.assertEqual ae(self.sris.size, 26) ae(self.sris._names, self.names) # end def test__init__ def test_seek(self): ae = self.assertEqual ar = self.assertRaises sris = self.sris ae(sris.seek(10, SEEK_SET), 10) ae(sris._position, 10) ar(ValueError, sris.seek, -10, SEEK_SET) sris.seek(3, SEEK_SET) ae(sris.seek(5, SEEK_CUR), 8) ae(sris._position, 8) ae(sris.seek(-2, SEEK_CUR), 6) ae(sris._position, 6) ae(sris.seek(-3, SEEK_END), 23) ae(sris._position, 23) ae(sris.seek(3, SEEK_END), 29) ae(sris._position, 29) # end def test_seek def test_tell(self): ae = self.assertEqual sris = self.sris sris._position = 0 ae(sris.tell(), 0) sris._position = 2 ae(sris.tell(), 2) # end def test_tell def test_read(self): ae = self.assertEqual self.sris.seek(0, SEEK_SET) ae(self.sris.read(0), b"") ae(self.sris.read(1), b"a") ae(self.sris.read(2), b"bc") ae(self.sris.read(), b"defghijklmnopqrstuvwxyz") self.sris.seek(-3, SEEK_END) ae(self.sris.read(5), b"xyz") self.sris.seek(30, SEEK_SET) ae(self.sris.read(), b"") # end def test_read def test_readall(self): ae = self.assertEqual self.sris.seek(0, SEEK_SET) ae(self.sris.readall(), b"abcdefghijklmnopqrstuvwxyz") self.sris.seek(3, SEEK_SET) ae(self.sris.readall(), b"defghijklmnopqrstuvwxyz") # end def test_readall def test_readinto(self): ae = self.assertEqual sris = self.sris barray0 = bytearray(5) barray1 = bytearray(10) barray2 = bytearray(26) barray3 = bytearray(1) sris.seek(-12, SEEK_END) retval0 = sris.readinto(barray0) retval1 = sris.readinto(barray1) sris.seek(0, SEEK_SET) retval2 = sris.readinto(barray2) sris.seek(30, SEEK_SET) retval3 = sris.readinto(barray3) ae(retval0, 5) ae(retval1, 7) ae(retval2, 26) ae(retval3, 0) ae(barray0, b"opqrs") ae(barray1, b"tuvwxyz\x00\x00\x00") ae(barray2, b"abcdefghijklmnopqrstuvwxyz") ae(barray3, b"\x00")
class SplitRawIStreamTestCase(TestCase): def setUp(self): names = ["alpha{0}.txt".format(x) for x in range(1, 7)] names = [os.path.join("data", "txt", name) for name in names] self.names = names self.sris = SplitRawIStream(names) # end def setUp def test__init__(self): ae = self.assertEqual ae(self.sris.size, 26) ae(self.sris._names, self.names) # end def test__init__ def test_seek(self): ae = self.assertEqual ar = self.assertRaises sris = self.sris ae(sris.seek(10, SEEK_SET), 10) ae(sris._position, 10) ar(ValueError, sris.seek, -10, SEEK_SET) sris.seek(3, SEEK_SET) ae(sris.seek(5, SEEK_CUR), 8) ae(sris._position, 8) ae(sris.seek(-2, SEEK_CUR), 6) ae(sris._position, 6) ae(sris.seek(-3, SEEK_END), 23) ae(sris._position, 23) ae(sris.seek(3, SEEK_END), 29) ae(sris._position, 29) # end def test_seek def test_tell(self): ae = self.assertEqual sris = self.sris sris._position = 0 ae(sris.tell(), 0) sris._position = 2 ae(sris.tell(), 2) # end def test_tell def test_read(self): ae = self.assertEqual self.sris.seek(0, SEEK_SET) ae(self.sris.read(0), b"") ae(self.sris.read(1), b"a") ae(self.sris.read(2), b"bc") ae(self.sris.read(), b"defghijklmnopqrstuvwxyz") self.sris.seek(-3, SEEK_END) ae(self.sris.read(5), b"xyz") self.sris.seek(30, SEEK_SET) ae(self.sris.read(), b"") # end def test_read def test_readall(self): ae = self.assertEqual self.sris.seek(0, SEEK_SET) ae(self.sris.readall(), b"abcdefghijklmnopqrstuvwxyz") self.sris.seek(3, SEEK_SET) ae(self.sris.readall(), b"defghijklmnopqrstuvwxyz") # end def test_readall def test_readinto(self): ae = self.assertEqual sris = self.sris barray0 = bytearray(5) barray1 = bytearray(10) barray2 = bytearray(26) barray3 = bytearray(1) sris.seek(-12, SEEK_END) retval0 = sris.readinto(barray0) retval1 = sris.readinto(barray1) sris.seek(0, SEEK_SET) retval2 = sris.readinto(barray2) sris.seek(30, SEEK_SET) retval3 = sris.readinto(barray3) ae(retval0, 5) ae(retval1, 7) ae(retval2, 26) ae(retval3, 0) ae(barray0, b"opqrs") ae(barray1, b"tuvwxyz\x00\x00\x00") ae(barray2, b"abcdefghijklmnopqrstuvwxyz") ae(barray3, b"\x00")