File manager - Edit - /usr/local/lib/python3.9/dist-packages/pandas/tests/frame/test_arrow_interface.py
Back
import ctypes import pytest import pandas.util._test_decorators as td import pandas as pd pa = pytest.importorskip("pyarrow") @td.skip_if_no("pyarrow", min_version="14.0") def test_dataframe_arrow_interface(): df = pd.DataFrame({"a": [1, 2, 3], "b": ["a", "b", "c"]}) capsule = df.__arrow_c_stream__() assert ( ctypes.pythonapi.PyCapsule_IsValid( ctypes.py_object(capsule), b"arrow_array_stream" ) == 1 ) table = pa.table(df) expected = pa.table({"a": [1, 2, 3], "b": ["a", "b", "c"]}) assert table.equals(expected) schema = pa.schema([("a", pa.int8()), ("b", pa.string())]) table = pa.table(df, schema=schema) expected = expected.cast(schema) assert table.equals(expected) @td.skip_if_no("pyarrow", min_version="15.0") def test_dataframe_to_arrow(): df = pd.DataFrame({"a": [1, 2, 3], "b": ["a", "b", "c"]}) table = pa.RecordBatchReader.from_stream(df).read_all() expected = pa.table({"a": [1, 2, 3], "b": ["a", "b", "c"]}) assert table.equals(expected) schema = pa.schema([("a", pa.int8()), ("b", pa.string())]) table = pa.RecordBatchReader.from_stream(df, schema=schema).read_all() expected = expected.cast(schema) assert table.equals(expected)
| ver. 1.4 |
Github
|
.
| PHP 7.4.33 | Generation time: 0.38 |
proxy
|
phpinfo
|
Settings