class TestCustomBytesIO(unittest.TestCase): def setUp(self): self.instance = CustomBytesIO() def test_writable(self): assert hasattr(self.instance, 'write') assert self.instance.write(b'example') == 7 def test_readable(self): assert hasattr(self.instance, 'read') assert self.instance.read() == b'' assert self.instance.read(10) == b'' def test_can_read_after_writing_to(self): self.instance.write(b'example text') self.instance.read() == b'example text' def test_can_read_some_after_writing_to(self): self.instance.write(b'example text') self.instance.read(6) == b'exampl' def test_can_get_length(self): self.instance.write(b'example') self.instance.seek(0, 0) assert len(self.instance) == 7 def test_truncates_intelligently(self): self.instance.write(b'abcdefghijklmnopqrstuvwxyzabcd') # 30 bytes assert self.instance.tell() == 30 self.instance.seek(-10, 2) self.instance.smart_truncate() assert len(self.instance) == 10 assert self.instance.read() == b'uvwxyzabcd' assert self.instance.tell() == 10
def setUp(self): self.instance = CustomBytesIO()
class TestCustomBytesIO(unittest.TestCase): def setUp(self): self.instance = CustomBytesIO() def test_writable(self): assert hasattr(self.instance, 'write') assert self.instance.write(b'example') == 7 def test_readable(self): assert hasattr(self.instance, 'read') assert self.instance.read() == b'' assert self.instance.read(10) == b'' def test_can_read_after_writing_to(self): self.instance.write(b'example text') self.instance.read() == b'example text' def test_can_read_some_after_writing_to(self): self.instance.write(b'example text') self.instance.read(6) == b'exampl' def test_can_get_length(self): self.instance.write(b'example') self.instance.seek(0, 0) assert len(self.instance) == 7