Skip to content

Commit

Permalink
Add pyarrow usage test case
Browse files Browse the repository at this point in the history
  • Loading branch information
ssyssy committed Apr 29, 2024
1 parent f6cea96 commit a378719
Show file tree
Hide file tree
Showing 2 changed files with 48 additions and 0 deletions.
1 change: 1 addition & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
"pytest",
"pytest-aiohttp",
"ray",
"pyarrow",
]
},
python_requires=">=3.8",
Expand Down
47 changes: 47 additions & 0 deletions tests/fs/test_docker_pyarrow_usage.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
import logging
import os

from pyarrow.fs import FSSpecHandler
from pyarrow.fs import PyFileSystem

from alluxiofs import AlluxioFileSystem
from tests.conftest import ALLUXIO_FILE_PATH
from tests.conftest import LOCAL_FILE_PATH
from tests.fs.test_docker_fsspec_cat import ALLUXIO_PREFIX
from tests.fs.test_docker_fsspec_cat import FILE_PATH

LOGGER = logging.getLogger(__name__)


def alluxio_pyarrow_test(py_fs, alluxio_path, local_path):
file_size = os.path.getsize(local_path)

file_info = py_fs.get_file_info(alluxio_path)
assert file_info.is_file
assert file_info.size == file_size
assert file_info.path == alluxio_path

with py_fs.open_input_file(alluxio_path) as f:
alluxio_file_data = f.read()

with open(local_path, "rb") as local_file:
local_file_data = local_file.read()
assert local_file_data == alluxio_file_data


def test_alluxio_pyarrow(alluxio_file_system: AlluxioFileSystem):
py_fs = PyFileSystem(FSSpecHandler(alluxio_file_system))

alluxio_pyarrow_test(py_fs, ALLUXIO_FILE_PATH, LOCAL_FILE_PATH)
alluxio_pyarrow_test(
py_fs,
ALLUXIO_PREFIX + ALLUXIO_FILE_PATH,
LOCAL_FILE_PATH,
)
alluxio_pyarrow_test(py_fs, FILE_PATH, LOCAL_FILE_PATH)


def test_etcd_alluxio_pyarrow(
etcd_alluxio_file_system: AlluxioFileSystem,
):
test_alluxio_pyarrow(etcd_alluxio_file_system)

0 comments on commit a378719

Please sign in to comment.