Fix unit tests

Change-Id: I38b1e570c69af59aac917e96845b7add947d7196
This commit is contained in:
Wes McKinney committed 2017-05-13 16:01:54 -04:00
1 parent 10837a65d9
commit 22346d47d5
2 files changed
+3 -3

No files matched your search

+1 -1
View File
@@ -84,6 +84,6 @@ class BatchFileWriter(lib._BatchFileWriter):
StreamWriter = BatchStreamWriter
StreamReader = BatchStreamWriter
StreamReader = BatchStreamReader
FileWriter = BatchFileWriter
FileReader = BatchFileReader
+2 -2
View File
@@ -99,12 +99,12 @@ class TestFile(MessagingTest, unittest.TestCase):
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)