Skip to content

mock_loaders

ecoscope.platform.mock_loaders

Mock data loaders for wt-task testing.

This module provides a parquet loader registered via the wt_task.mock_loaders entry point group, enabling wt_task.testing.create_task_magicmock to load .parquet example-return files without wt-task depending on geopandas/pandas.

Functions:

load_parquet 

load_parquet(path: Path) -> Any

Load a parquet or geoparquet file.

Attempts to read as a GeoDataFrame first (via geopandas); falls back to a plain DataFrame (via pandas) if the file doesn't contain geometry data.

Parameters:

Name Type Description Default
path Path

Path to the parquet file.

 required

Returns:

Type Description
Any

A GeoDataFrame or DataFrame with the file contents.
Source code in ecoscope/platform/mock_loaders.py 
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
def load_parquet(path: Path) -> Any:
    """Load a parquet or geoparquet file.

    Attempts to read as a GeoDataFrame first (via geopandas); falls back to
    a plain DataFrame (via pandas) if the file doesn't contain geometry data.

    Args:
        path: Path to the parquet file.

    Returns:
        A GeoDataFrame or DataFrame with the file contents.
    """
    import geopandas as gpd  # type: ignore[import-untyped]
    import pandas as pd

    try:
        return gpd.read_parquet(path.as_uri())
    except ValueError:
        return pd.read_parquet(path.as_uri())