81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
"""
|
||
tests/conftest.py
|
||
~~~~~~~~~~~~~~~~~
|
||
共享测试夹具（Fixture）—— 提供测试数据，供所有测试文件复用。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import tempfile
|
||
from pathlib import Path
|
||
|
||
import geopandas as gpd
|
||
import pytest
|
||
from shapely.geometry import Point, Polygon
|
||
|
||
|
||
# ── 示例 GeoDataFrame ──────────────────────────────────────────────────────────
|
||
|
||
@pytest.fixture
|
||
def sample_points_gdf() -> gpd.GeoDataFrame:
|
||
"""3 个点的 GeoDataFrame（WGS84）。"""
|
||
return gpd.GeoDataFrame(
|
||
{
|
||
"id": [1, 2, 3],
|
||
"name": ["点A", "点B", "点C"],
|
||
"value": [10.5, 20.0, 15.3],
|
||
},
|
||
geometry=[Point(116.4, 39.9), Point(121.5, 31.2), Point(113.3, 23.1)],
|
||
crs="EPSG:4326",
|
||
)
|
||
|
||
|
||
@pytest.fixture
|
||
def sample_polygon_gdf() -> gpd.GeoDataFrame:
|
||
"""1 个矩形多边形的 GeoDataFrame（WGS84）。"""
|
||
poly = Polygon([(115.0, 38.0), (122.0, 38.0), (122.0, 41.0), (115.0, 41.0)])
|
||
return gpd.GeoDataFrame(
|
||
{"region": ["华北区"], "area_km2": [450000.0]},
|
||
geometry=[poly],
|
||
crs="EPSG:4326",
|
||
)
|
||
|
||
|
||
@pytest.fixture
|
||
def sample_multi_polygon_gdf() -> gpd.GeoDataFrame:
|
||
"""包含两个多边形的 GeoDataFrame，用于融合/叠置测试（WGS84）。"""
|
||
poly1 = Polygon([(100, 20), (110, 20), (110, 30), (100, 30)])
|
||
poly2 = Polygon([(105, 20), (115, 20), (115, 30), (105, 30)])
|
||
return gpd.GeoDataFrame(
|
||
{"zone": ["A", "B"], "value": [100, 200]},
|
||
geometry=[poly1, poly2],
|
||
crs="EPSG:4326",
|
||
)
|
||
|
||
|
||
# ── 临时文件路径 ───────────────────────────────────────────────────────────────
|
||
|
||
@pytest.fixture
|
||
def tmp_geojson_path(tmp_path: Path, sample_points_gdf: gpd.GeoDataFrame) -> Path:
|
||
"""将 sample_points_gdf 写出为临时 GeoJSON 并返回路径。"""
|
||
path = tmp_path / "sample.geojson"
|
||
sample_points_gdf.to_file(str(path), driver="GeoJSON")
|
||
return path
|
||
|
||
|
||
@pytest.fixture
|
||
def tmp_gpkg_path(tmp_path: Path, sample_points_gdf: gpd.GeoDataFrame) -> Path:
|
||
"""将 sample_points_gdf 写出为临时 GPKG 并返回路径。"""
|
||
path = tmp_path / "sample.gpkg"
|
||
sample_points_gdf.to_file(str(path), driver="GPKG", layer="points")
|
||
return path
|
||
|
||
|
||
@pytest.fixture
|
||
def tmp_output_dir(tmp_path: Path) -> Path:
|
||
"""空的临时输出目录。"""
|
||
out = tmp_path / "output"
|
||
out.mkdir()
|
||
return out
|