Add ci for python tests

Author: beinan

.github/workflows/python-test.yml

@@ -0,0 +1,83 @@
+name: Python Tests
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    paths:
+      - python/**
+      - rust/lance-graph/**
+      - .github/workflows/python-test.yml
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: true
+
+env:
+  CARGO_TERM_COLOR: always
+  RUSTFLAGS: "-C debuginfo=1"
+
+jobs:
+  test:
+    runs-on: ubuntu-24.04
+    timeout-minutes: 30
+    strategy:
+      matrix:
+        python-version: ["3.11"]
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Setup Rust
+        uses: actions-rust-lang/setup-rust-toolchain@v1
+      - uses: Swatinem/rust-cache@v2
+        with:
+          workspaces: python
+      - name: Install dependencies
+        run: |
+          sudo apt update
+          sudo apt install -y protobuf-compiler
+      - name: Install Python dependencies
+        working-directory: python
+        run: |
+          python -m pip install --upgrade pip
+          pip install maturin[patchelf]
+          pip install -e .[tests]
+      - name: Build Python extension
+        working-directory: python
+        run: maturin develop
+      - name: Run tests
+        working-directory: python
+        run: pytest python/tests/ -v
+      - name: Run doctests
+        working-directory: python
+        run: |
+          if [ -f python/lance_graph/__init__.py ]; then
+            python -m doctest python/lance_graph/__init__.py || echo "No doctests found"
+          fi
+
+  lint:
+    runs-on: ubuntu-24.04
+    timeout-minutes: 15
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.11"
+      - name: Install dependencies
+        working-directory: python
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e .[dev]
+      - name: Run ruff format check
+        working-directory: python
+        run: ruff format --check python/
+      - name: Run ruff lint
+        working-directory: python
+        run: ruff check python/
+      - name: Run pyright type check
+        working-directory: python
+        run: pyright

python/python/lance_graph/__init__.py

@@ -5,7 +5,10 @@
 import importlib.util
 import sys
 from pathlib import Path
-from types import ModuleType
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from types import ModuleType
 
 
 def _load_bindings() -> ModuleType:

python/python/tests/test_graph.py

@@ -3,7 +3,6 @@
 
 import pyarrow as pa
 import pytest
-
 from lance_graph import CypherQuery, GraphConfig