Fix unit tests

Change-Id: I38b1e570c69af59aac917e96845b7add947d7196
wesm · May 13, 2017 · 22346d4 · 22346d4
1 parent 10837a6
commit 22346d4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/pyarrow/ipc.py b/python/pyarrow/ipc.py
@@ -84,6 +84,6 @@ def __init__(self, sink, schema):
 
 
 StreamWriter = BatchStreamWriter
-StreamReader = BatchStreamWriter
+StreamReader = BatchStreamReader
 FileWriter = BatchFileWriter
 FileReader = BatchFileReader
diff --git a/python/pyarrow/tests/test_ipc.py b/python/pyarrow/tests/test_ipc.py
@@ -99,12 +99,12 @@ def test_read_all(self):
 class TestStream(MessagingTest, unittest.TestCase):
 
     def _get_writer(self, sink, schema):
-        return pa.StreamWriter(sink, schema)
+        return pa.BatchStreamWriter(sink, schema)
 
     def test_simple_roundtrip(self):
         batches = self.write_batches()
         file_contents = self._get_source()
-        reader = pa.StreamReader(file_contents)
+        reader = pa.BatchStreamReader(file_contents)
 
         assert reader.schema.equals(batches[0].schema)