X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/2d2d6ad747ccb94d77fcb11c275317af7c08f7af..a2c147e469bc4ccc0e788f2e72b2d94aa81ba368:/sdk/python/tests/test_stream.py?ds=sidebyside diff --git a/sdk/python/tests/test_stream.py b/sdk/python/tests/test_stream.py index 2c31ef9954..624f1b8ca4 100644 --- a/sdk/python/tests/test_stream.py +++ b/sdk/python/tests/test_stream.py @@ -191,6 +191,12 @@ class StreamFileReaderTestCase(unittest.TestCase): data = reader.readlines() self.assertEqual(['two\n', '\n', 'three\n', 'four\n', '\n'], data) + def test_readline_then_readall(self): + reader = self.make_newlines_reader() + data = reader.readline() + self.assertEqual('one\n', data) + self.assertEqual(''.join(['two\n', '\n', 'three\n', 'four\n', '\n']), ''.join(reader.readall())) + class StreamRetryTestMixin(object): # Define reader_for(coll_name, **kwargs)