mirror of
https://github.com/lancedb/lancedb.git
synced 2026-01-07 12:22:59 +00:00
82b25a71e9
Add a new test feature which allows for running the lancedb tests against a remote server. Convert over a few tests in src/connection.rs as a proof of concept. To make local development easier, the remote tests can be run locally from a Makefile. This file can also be used to run the feature tests, with a single invocation of 'make'. (The feature tests require bringing up a docker compose environment.)
52 lines
1.2 KiB
Bash
Executable File
52 lines
1.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
#
|
|
# A script for running the given command together with the lancedb cli.
|
|
#
|
|
|
|
die() {
|
|
echo $?
|
|
exit 1
|
|
}
|
|
|
|
check_command_exists() {
|
|
command="${1}"
|
|
which ${command} &> /dev/null || \
|
|
die "Unable to locate command: ${command}. Did you install it?"
|
|
}
|
|
|
|
if [[ ! -e ./lancedb ]]; then
|
|
ARCH="x64"
|
|
if [[ $OSTYPE == 'darwin'* ]]; then
|
|
UNAME=$(uname -m)
|
|
if [[ $UNAME == 'arm64' ]]; then
|
|
ARCH='arm64'
|
|
fi
|
|
OSTYPE="macos"
|
|
elif [[ $OSTYPE == 'linux'* ]]; then
|
|
if [[ $UNAME == 'aarch64' ]]; then
|
|
ARCH='arm64'
|
|
fi
|
|
OSTYPE="linux"
|
|
else
|
|
die "unknown OSTYPE: $OSTYPE"
|
|
fi
|
|
|
|
check_command_exists gh
|
|
TARGET="lancedb-${OSTYPE}-${ARCH}.tar.gz"
|
|
gh release \
|
|
--repo lancedb/sophon \
|
|
download lancedb-cli-v0.0.3 \
|
|
--pattern "${TARGET}" \
|
|
|| die "failed to fetch cli."
|
|
|
|
check_command_exists tar
|
|
tar xvf "${TARGET}" || die "tar failed."
|
|
[[ -e ./lancedb ]] || die "failed to extract lancedb."
|
|
fi
|
|
|
|
SCRIPT_DIR=$(dirname "$(readlink -f "$0")")
|
|
export CREATE_LANCEDB_TEST_CONNECTION_SCRIPT="${SCRIPT_DIR}/create_lancedb_test_connection.sh"
|
|
|
|
"${@}"
|