11308: Futurize stage2.

[arvados.git] / sdk / python / tests / test_stream.py

diff --git a/sdk/python/tests/test_stream.py b/sdk/python/tests/test_stream.py
index 2c31ef99541385f98082b3cc0c727d262e5b9c09..7277628122f31205dd477b20ad593c25a294df46 100644 (file)
--- a/sdk/python/tests/test_stream.py
+++ b/sdk/python/tests/test_stream.py
@@ -1,5 +1,7 @@
 #!/usr/bin/env python
 
+from __future__ import absolute_import
+from builtins import object
 import bz2
 import gzip
 import io
@@ -12,8 +14,8 @@ import arvados
 from arvados import StreamReader, StreamFileReader
 from arvados._ranges import Range
 
-import arvados_testutil as tutil
-import run_test_server
+from . import arvados_testutil as tutil
+from . import run_test_server
 
 class StreamFileReaderTestCase(unittest.TestCase):
     def make_count_reader(self):
@@ -191,6 +193,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)