mirror of
https://github.com/apache/arrow.git
synced 2026-10-29 17:15:39 +08:00
Fix unit tests
Change-Id: I38b1e570c69af59aac917e96845b7add947d7196
This commit is contained in:
1 parent
10837a65d9
commit
22346d47d5
2 files changed
+3
-3
No files matched your search
@@ -84,6 +84,6 @@ class BatchFileWriter(lib._BatchFileWriter):
|
||||
|
||||
|
||||
StreamWriter = BatchStreamWriter
|
||||
StreamReader = BatchStreamWriter
|
||||
StreamReader = BatchStreamReader
|
||||
FileWriter = BatchFileWriter
|
||||
FileReader = BatchFileReader
|
||||
@@ -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)
|
||||
|
||||
|
||||
Reference in new issue
Block a user