mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-07-04 18:23:07 +02:00
Krzysztof kuhy Rudnicki
ee27d10fef
Fix ruff violations in ~15 source files and ~60+ test files to minimize per-file-ignores in pyproject.toml. Remaining ignores are justified with comments explaining why each suppression is necessary. Source fixes: FBT003 (keyword args), S310 (URL validation), SLF001 (private access), T201 (print→logging), C901 (complexity), E501 (line length), E402 (import order). Test fixes: SIM117 (combined with), FBT (boolean args), PERF203 (try in loop), S310/S607 (URLs/executables), E402/E501 (imports/lines), S108 (tmp paths), PLR0913 (too many args), ARG (unused args), ANN (type annotations), RUF059 (unused unpacked vars), PT019 (fixture naming). Remaining per-file-ignores (with justifications): - Tests: ARG, D, PLC0415, PLR2004, S101, SLF001 - music_gen sources: PLC0415 (heavy ML lazy imports) - moviepy_showcase: PLC0415 (circular dependency) - generate_images: PLR0913 (matplotlib helpers need many params) - praca_magisterska_video: E501, E402 (long paths, mpl.use)
419 lines
16 KiB
Python
419 lines
16 KiB
Python
"""Tests for forests, nature reserves, and landscape parks download paths."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import geopandas as gpd
|
|
from shapely.geometry import Polygon
|
|
|
|
from python_pkg.geo_data._poland_nature import (
|
|
get_polish_forests,
|
|
get_polish_landscape_parks,
|
|
get_polish_nature_reserves,
|
|
)
|
|
|
|
|
|
def _make_relation_element(name: str, *, include_outer: bool = True) -> dict[str, Any]:
|
|
"""Create a mock OSM relation element."""
|
|
members = []
|
|
if include_outer:
|
|
members.append(
|
|
{
|
|
"role": "outer",
|
|
"geometry": [
|
|
{"lon": 0, "lat": 0},
|
|
{"lon": 1, "lat": 0},
|
|
{"lon": 1, "lat": 1},
|
|
{"lon": 0, "lat": 1},
|
|
],
|
|
}
|
|
)
|
|
return {"type": "relation", "tags": {"name": name}, "members": members}
|
|
|
|
|
|
_POLY = Polygon([(20, 50), (21, 50), (21, 51), (20, 51)])
|
|
|
|
|
|
class TestGetPolishForests:
|
|
"""Tests for get_polish_forests."""
|
|
|
|
@patch("python_pkg.geo_data._poland_nature.gpd.read_file")
|
|
@patch("python_pkg.geo_data._poland_nature.CACHE_DIR")
|
|
def test_cached_with_area(
|
|
self, mock_cache_dir: MagicMock, mock_read: MagicMock
|
|
) -> None:
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = True
|
|
mock_gdf = gpd.GeoDataFrame(
|
|
{"name": ["Puszcza Białowieska"], "area_km2": [600.0]},
|
|
geometry=[_POLY],
|
|
crs="EPSG:4326",
|
|
)
|
|
mock_read.return_value = mock_gdf
|
|
result = get_polish_forests()
|
|
assert result.iloc[0]["area_km2"] == 600.0
|
|
|
|
@patch("python_pkg.geo_data._poland_nature.gpd.read_file")
|
|
@patch("python_pkg.geo_data._poland_nature.CACHE_DIR")
|
|
def test_cached_without_area(
|
|
self, mock_cache_dir: MagicMock, mock_read: MagicMock
|
|
) -> None:
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = True
|
|
mock_gdf = gpd.GeoDataFrame(
|
|
{"name": ["Puszcza Białowieska"]},
|
|
geometry=[_POLY],
|
|
crs="EPSG:4326",
|
|
)
|
|
mock_read.return_value = mock_gdf
|
|
result = get_polish_forests()
|
|
assert len(result) == 1
|
|
|
|
def test_downloads_forests(self) -> None:
|
|
with (
|
|
patch("python_pkg.geo_data._poland_nature.sys.stdout"),
|
|
patch("python_pkg.geo_data._poland_nature.CACHE_DIR") as mock_cache_dir,
|
|
patch("python_pkg.geo_data._poland_nature._overpass_query") as mock_query,
|
|
patch("python_pkg.geo_data._poland_nature._ensure_cache_dir"),
|
|
patch(
|
|
"python_pkg.geo_data._poland_nature.gpd.GeoDataFrame.from_features"
|
|
) as mock_from_features,
|
|
patch(
|
|
"python_pkg.geo_data._poland_nature._add_area_column"
|
|
) as mock_add_area,
|
|
):
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = False
|
|
|
|
mock_query.return_value = {
|
|
"elements": [
|
|
# Valid forest with keyword
|
|
{
|
|
"type": "way",
|
|
"tags": {"name": "Puszcza Białowieska"},
|
|
"geometry": [
|
|
{"lon": 0, "lat": 0},
|
|
{"lon": 1, "lat": 0},
|
|
{"lon": 1, "lat": 1},
|
|
{"lon": 0, "lat": 1},
|
|
],
|
|
},
|
|
# Bory keyword
|
|
{
|
|
"type": "way",
|
|
"tags": {"name": "Bory Tucholskie"},
|
|
"geometry": [
|
|
{"lon": 2, "lat": 2},
|
|
{"lon": 3, "lat": 2},
|
|
{"lon": 3, "lat": 3},
|
|
{"lon": 2, "lat": 3},
|
|
],
|
|
},
|
|
# No forest keyword -> skip
|
|
{
|
|
"type": "way",
|
|
"tags": {"name": "Random Wood"},
|
|
"geometry": [
|
|
{"lon": 0, "lat": 0},
|
|
{"lon": 1, "lat": 0},
|
|
{"lon": 1, "lat": 1},
|
|
{"lon": 0, "lat": 1},
|
|
],
|
|
},
|
|
# Duplicate
|
|
{
|
|
"type": "way",
|
|
"tags": {"name": "Puszcza Białowieska"},
|
|
"geometry": [
|
|
{"lon": 0, "lat": 0},
|
|
{"lon": 1, "lat": 0},
|
|
{"lon": 1, "lat": 1},
|
|
{"lon": 0, "lat": 1},
|
|
],
|
|
},
|
|
# No name
|
|
{"type": "way", "tags": {}, "geometry": []},
|
|
# Geometry extraction fails (too few coords)
|
|
{
|
|
"type": "way",
|
|
"tags": {"name": "Las Mały"},
|
|
"geometry": [{"lon": 0, "lat": 0}],
|
|
},
|
|
]
|
|
}
|
|
|
|
mock_gdf = gpd.GeoDataFrame(
|
|
{"name": ["Puszcza Białowieska", "Bory Tucholskie"]},
|
|
geometry=[_POLY, _POLY],
|
|
crs="EPSG:4326",
|
|
)
|
|
mock_from_features.return_value = mock_gdf
|
|
gdf_with_area = mock_gdf.copy()
|
|
gdf_with_area["area_km2"] = [600.0, 300.0]
|
|
mock_add_area.return_value = gdf_with_area
|
|
|
|
result = get_polish_forests()
|
|
assert len(result) == 2
|
|
|
|
def test_downloads_forests_empty(self) -> None:
|
|
with (
|
|
patch("python_pkg.geo_data._poland_nature.sys.stdout"),
|
|
patch("python_pkg.geo_data._poland_nature.CACHE_DIR") as mock_cache_dir,
|
|
patch("python_pkg.geo_data._poland_nature._overpass_query") as mock_query,
|
|
patch("python_pkg.geo_data._poland_nature._ensure_cache_dir"),
|
|
patch(
|
|
"python_pkg.geo_data._poland_nature.gpd.GeoDataFrame.from_features"
|
|
) as mock_from_features,
|
|
patch(
|
|
"python_pkg.geo_data._poland_nature._add_area_column"
|
|
) as mock_add_area,
|
|
):
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = False
|
|
mock_query.return_value = {"elements": []}
|
|
empty_gdf = gpd.GeoDataFrame({"name": [], "geometry": []})
|
|
mock_from_features.return_value = empty_gdf
|
|
mock_add_area.return_value = empty_gdf
|
|
result = get_polish_forests()
|
|
assert len(result) == 0
|
|
|
|
|
|
class TestGetPolishNatureReserves:
|
|
"""Tests for get_polish_nature_reserves."""
|
|
|
|
@patch("python_pkg.geo_data._poland_nature.gpd.read_file")
|
|
@patch("python_pkg.geo_data._poland_nature.CACHE_DIR")
|
|
def test_cached_with_area(
|
|
self, mock_cache_dir: MagicMock, mock_read: MagicMock
|
|
) -> None:
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = True
|
|
mock_gdf = gpd.GeoDataFrame(
|
|
{"name": ["Rezerwat X"], "area_km2": [50.0]},
|
|
geometry=[_POLY],
|
|
crs="EPSG:4326",
|
|
)
|
|
mock_read.return_value = mock_gdf
|
|
result = get_polish_nature_reserves()
|
|
assert result.iloc[0]["area_km2"] == 50.0
|
|
|
|
@patch("python_pkg.geo_data._poland_nature.gpd.read_file")
|
|
@patch("python_pkg.geo_data._poland_nature.CACHE_DIR")
|
|
def test_cached_without_area(
|
|
self, mock_cache_dir: MagicMock, mock_read: MagicMock
|
|
) -> None:
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = True
|
|
mock_gdf = gpd.GeoDataFrame(
|
|
{"name": ["Rezerwat X"]},
|
|
geometry=[_POLY],
|
|
crs="EPSG:4326",
|
|
)
|
|
mock_read.return_value = mock_gdf
|
|
result = get_polish_nature_reserves()
|
|
assert len(result) == 1
|
|
|
|
def test_downloads_reserves(self) -> None:
|
|
with (
|
|
patch("python_pkg.geo_data._poland_nature.sys.stdout"),
|
|
patch("python_pkg.geo_data._poland_nature.CACHE_DIR") as mock_cache_dir,
|
|
patch("python_pkg.geo_data._poland_nature._overpass_query") as mock_query,
|
|
patch("python_pkg.geo_data._poland_nature._ensure_cache_dir"),
|
|
patch(
|
|
"python_pkg.geo_data._poland_nature.gpd.GeoDataFrame.from_features"
|
|
) as mock_from_features,
|
|
patch(
|
|
"python_pkg.geo_data._poland_nature._add_area_column"
|
|
) as mock_add_area,
|
|
):
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = False
|
|
|
|
mock_query.return_value = {
|
|
"elements": [
|
|
{
|
|
"type": "way",
|
|
"tags": {"name": "Rezerwat A"},
|
|
"geometry": [
|
|
{"lon": 0, "lat": 0},
|
|
{"lon": 1, "lat": 0},
|
|
{"lon": 1, "lat": 1},
|
|
{"lon": 0, "lat": 1},
|
|
],
|
|
},
|
|
# Duplicate
|
|
{
|
|
"type": "way",
|
|
"tags": {"name": "Rezerwat A"},
|
|
"geometry": [
|
|
{"lon": 0, "lat": 0},
|
|
{"lon": 1, "lat": 0},
|
|
{"lon": 1, "lat": 1},
|
|
{"lon": 0, "lat": 1},
|
|
],
|
|
},
|
|
# No name
|
|
{"type": "way", "tags": {}, "geometry": []},
|
|
# Geometry fails
|
|
{
|
|
"type": "way",
|
|
"tags": {"name": "Tiny"},
|
|
"geometry": [{"lon": 0, "lat": 0}],
|
|
},
|
|
]
|
|
}
|
|
|
|
mock_gdf = gpd.GeoDataFrame(
|
|
{"name": ["Rezerwat A"]},
|
|
geometry=[_POLY],
|
|
crs="EPSG:4326",
|
|
)
|
|
mock_from_features.return_value = mock_gdf
|
|
gdf_with_area = mock_gdf.copy()
|
|
gdf_with_area["area_km2"] = [50.0]
|
|
mock_add_area.return_value = gdf_with_area
|
|
|
|
result = get_polish_nature_reserves()
|
|
assert len(result) == 1
|
|
|
|
def test_downloads_reserves_empty(self) -> None:
|
|
with (
|
|
patch("python_pkg.geo_data._poland_nature.sys.stdout"),
|
|
patch("python_pkg.geo_data._poland_nature.CACHE_DIR") as mock_cache_dir,
|
|
patch("python_pkg.geo_data._poland_nature._overpass_query") as mock_query,
|
|
patch("python_pkg.geo_data._poland_nature._ensure_cache_dir"),
|
|
patch(
|
|
"python_pkg.geo_data._poland_nature.gpd.GeoDataFrame.from_features"
|
|
) as mock_from_features,
|
|
patch(
|
|
"python_pkg.geo_data._poland_nature._add_area_column"
|
|
) as mock_add_area,
|
|
):
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = False
|
|
mock_query.return_value = {"elements": []}
|
|
empty_gdf = gpd.GeoDataFrame({"name": [], "geometry": []})
|
|
mock_from_features.return_value = empty_gdf
|
|
mock_add_area.return_value = empty_gdf
|
|
result = get_polish_nature_reserves()
|
|
assert len(result) == 0
|
|
|
|
|
|
class TestGetPolishLandscapeParks:
|
|
"""Tests for get_polish_landscape_parks."""
|
|
|
|
@patch("python_pkg.geo_data._poland_nature.gpd.read_file")
|
|
@patch("python_pkg.geo_data._poland_nature.CACHE_DIR")
|
|
def test_cached_with_area(
|
|
self,
|
|
mock_cache_dir: MagicMock,
|
|
mock_read: MagicMock,
|
|
) -> None:
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = True
|
|
mock_gdf = gpd.GeoDataFrame(
|
|
{"name": ["Park Krajobrazowy X"], "area_km2": [100.0]},
|
|
geometry=[_POLY],
|
|
crs="EPSG:4326",
|
|
)
|
|
mock_read.return_value = mock_gdf
|
|
result = get_polish_landscape_parks()
|
|
assert result.iloc[0]["area_km2"] == 100.0
|
|
|
|
@patch("python_pkg.geo_data._poland_nature.gpd.read_file")
|
|
@patch("python_pkg.geo_data._poland_nature.CACHE_DIR")
|
|
def test_cached_without_area(
|
|
self,
|
|
mock_cache_dir: MagicMock,
|
|
mock_read: MagicMock,
|
|
) -> None:
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = True
|
|
mock_gdf = gpd.GeoDataFrame(
|
|
{"name": ["Park Krajobrazowy X"]},
|
|
geometry=[_POLY],
|
|
crs="EPSG:4326",
|
|
)
|
|
mock_read.return_value = mock_gdf
|
|
result = get_polish_landscape_parks()
|
|
assert len(result) == 1
|
|
|
|
@patch("python_pkg.geo_data._poland_nature.gpd.GeoDataFrame.from_features")
|
|
@patch("python_pkg.geo_data._poland_nature._ensure_cache_dir")
|
|
@patch("python_pkg.geo_data._poland_nature._overpass_query")
|
|
@patch("python_pkg.geo_data._poland_nature.CACHE_DIR")
|
|
@patch("python_pkg.geo_data._poland_nature.sys.stdout")
|
|
def test_downloads_landscape_parks(
|
|
self,
|
|
mock_stdout: MagicMock,
|
|
mock_cache_dir: MagicMock,
|
|
mock_query: MagicMock,
|
|
mock_ensure: MagicMock,
|
|
mock_from_features: MagicMock,
|
|
) -> None:
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = False
|
|
|
|
mock_query.return_value = {
|
|
"elements": [
|
|
_make_relation_element("Park Krajobrazowy A"),
|
|
# Not a relation -> skip
|
|
{
|
|
"type": "way",
|
|
"tags": {"name": "Park Krajobrazowy B"},
|
|
"geometry": [],
|
|
},
|
|
# No name
|
|
{"type": "relation", "tags": {}, "members": []},
|
|
# Duplicate
|
|
_make_relation_element("Park Krajobrazowy A"),
|
|
# No outer rings
|
|
_make_relation_element("Park Empty", include_outer=False),
|
|
]
|
|
}
|
|
|
|
mock_gdf = gpd.GeoDataFrame(
|
|
{"name": ["Park Krajobrazowy A"]},
|
|
geometry=[_POLY],
|
|
crs="EPSG:4326",
|
|
)
|
|
mock_from_features.return_value = mock_gdf
|
|
|
|
result = get_polish_landscape_parks()
|
|
assert len(result) == 1
|
|
|
|
@patch("python_pkg.geo_data._poland_nature.gpd.GeoDataFrame.from_features")
|
|
@patch("python_pkg.geo_data._poland_nature._ensure_cache_dir")
|
|
@patch("python_pkg.geo_data._poland_nature._overpass_query")
|
|
@patch("python_pkg.geo_data._poland_nature.CACHE_DIR")
|
|
@patch("python_pkg.geo_data._poland_nature.sys.stdout")
|
|
def test_downloads_landscape_parks_empty(
|
|
self,
|
|
mock_stdout: MagicMock,
|
|
mock_cache_dir: MagicMock,
|
|
mock_query: MagicMock,
|
|
mock_ensure: MagicMock,
|
|
mock_from_features: MagicMock,
|
|
) -> None:
|
|
mock_path = MagicMock()
|
|
mock_cache_dir.__truediv__ = MagicMock(return_value=mock_path)
|
|
mock_path.exists.return_value = False
|
|
mock_query.return_value = {"elements": []}
|
|
empty_gdf = gpd.GeoDataFrame({"name": [], "geometry": []})
|
|
mock_from_features.return_value = empty_gdf
|
|
result = get_polish_landscape_parks()
|
|
assert len(result) == 0
|