From f99bad0519cfb6b25730668fd98af6f8a416bf64 Mon Sep 17 00:00:00 2001 From: Magnus Ulimoen Date: Fri, 17 Jan 2025 15:29:48 +0100 Subject: [PATCH] Require merging mode to be specified --- src/pyaro_readers/merging_reader.py | 16 +++++++--------- tests/test_merginreader.py | 1 + 2 files changed, 8 insertions(+), 9 deletions(-) diff --git a/src/pyaro_readers/merging_reader.py b/src/pyaro_readers/merging_reader.py index 6d2a0bb..d95c0cd 100644 --- a/src/pyaro_readers/merging_reader.py +++ b/src/pyaro_readers/merging_reader.py @@ -1,15 +1,7 @@ -import logging -from datetime import datetime, timedelta -from pathlib import Path from typing import Literal, Any -from collections.abc import Iterable -import importlib.resources -import dataclasses -from tqdm import tqdm import numpy as np import cf_units -import polars from pyaro.timeseries.AutoFilterReaderEngine import ( AutoFilterReader, AutoFilterEngine, @@ -91,7 +83,13 @@ def __len__(self) -> int: class MergingReader(AutoFilterReader): - def __init__(self, datasets: list[dict[str, Any]], filters=[]): + def __init__( + self, datasets: list[dict[str, Any]], mode: Literal["concat"], filters=[] + ): + if mode != "concat": + raise MergingReaderException( + 'Only merging mode "concat" is supported as of now' + ) self._datasets = [] self._set_filters(filters) for d in datasets: diff --git a/tests/test_merginreader.py b/tests/test_merginreader.py index 4743503..32f54c5 100644 --- a/tests/test_merginreader.py +++ b/tests/test_merginreader.py @@ -14,6 +14,7 @@ def test_stuff(): {"readername": "ascii2netcdf", "filename": EBAS_URL}, {"readername": "ascii2netcdf", "filename": EBAS_URL}, ], + mode="concat", filters=[], ) as ts: ts.variables()