mirror of
https://github.com/apache/arrow.git
synced 2026-10-28 16:47:19 +08:00
Use PARQUET from miniconda path
This commit is contained in:
1 parent
cd3b9a9d42
commit
9520c39f48
2 files changed
+14
No files matched your search
@@ -7,6 +7,7 @@ PYTHON_DIR=$TRAVIS_BUILD_DIR/python
|
||||
# Re-use conda installation from C++
|
||||
export MINICONDA=$TRAVIS_BUILD_DIR/miniconda
|
||||
export PATH="$MINICONDA/bin:$PATH"
|
||||
export PARQUET_HOME=$MINICONDA
|
||||
|
||||
# Share environment with C++
|
||||
pushd $CPP_BUILD_DIR
|
||||
|
||||
@@ -27,6 +27,19 @@ from tempfile import mkdtemp
|
||||
import os.path
|
||||
|
||||
|
||||
def test_single_int64_column(tmpdir):
|
||||
filename = tmpdir.join('single_int64_column.parquet')
|
||||
data = [A.from_pylist(range(5))]
|
||||
table = A.Table.from_arrays(('a', 'b'), data, 'table_name')
|
||||
A.parquet.write_table(table, filename.strpath)
|
||||
table_read = pyarrow.parquet.read_table(filename.strpath)
|
||||
for col_written, col_read in zip(table.itercolumns(), table_read.itercolumns()):
|
||||
assert col_written.name == col_read.name
|
||||
assert col_read.data.num_chunks == 1
|
||||
data_written = col_written.data.chunk(0)
|
||||
data_read = col_read.data.chunk(0)
|
||||
assert data_written == data_read
|
||||
|
||||
class TestParquetIO(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
|
||||
Reference in new issue
Block a user