anemoi-datasets 0.3.3__tar.gz → 0.3.5__tar.gz

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/.github/workflows/python-publish.yml

@@ -42,8 +42,7 @@ jobs:
     - name: Install
       run: |
         pip install pytest
-        pip install -e .[all]
-        pip install -r tests/requirements.txt
+        pip install -e .[all,test]
         pip freeze
 
     - name: Tests

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/.pre-commit-config.yaml

@@ -1,17 +1,14 @@
 repos:
-
 # Empty notebookds
 - repo: local
   hooks:
-    - id: clear-notebooks-output
-      name: clear-notebooks-output
-      files: tools/.*\.ipynb$
-      stages: [commit]
-      language: python
-      entry: jupyter nbconvert --ClearOutputPreprocessor.enabled=True --inplace
-      additional_dependencies: [jupyter]
-
-
+  - id: clear-notebooks-output
+    name: clear-notebooks-output
+    files: tools/.*\.ipynb$
+    stages: [commit]
+    language: python
+    entry: jupyter nbconvert --ClearOutputPreprocessor.enabled=True --inplace
+    additional_dependencies: [jupyter]
 - repo: https://github.com/pre-commit/pre-commit-hooks
   rev: v4.6.0
   hooks:
@@ -23,13 +20,11 @@ repos:
   - id: no-commit-to-branch # Prevent committing to main / master
   - id: check-added-large-files # Check for large files added to git
   - id: check-merge-conflict # Check for files that contain merge conflict
-
 - repo: https://github.com/psf/black-pre-commit-mirror
   rev: 24.4.2
   hooks:
-    - id: black
-      args: [--line-length=120]
-
+  - id: black
+    args: [--line-length=120]
 - repo: https://github.com/pycqa/isort
   rev: 5.13.2
   hooks:
@@ -38,8 +33,6 @@ repos:
     - -l 120
     - --force-single-line-imports
     - --profile black
-
-
 - repo: https://github.com/astral-sh/ruff-pre-commit
   rev: v0.4.6
   hooks:
@@ -50,32 +43,27 @@ repos:
     - --fix
     - --exit-non-zero-on-fix
     - --preview
-
 - repo: https://github.com/sphinx-contrib/sphinx-lint
   rev: v0.9.1
   hooks:
-    - id: sphinx-lint
-
+  - id: sphinx-lint
 # For now, we use it. But it does not support a lot of sphinx features
 - repo: https://github.com/dzhu/rstfmt
   rev: v0.0.14
   hooks:
-    - id: rstfmt
-      exclude: 'cli/.*' # Because we use argparse
-
+  - id: rstfmt
+    exclude: 'cli/.*' # Because we use argparse
 - repo: https://github.com/b8raoult/pre-commit-docconvert
   rev: "0.1.5"
   hooks:
   - id: docconvert
     args: ["numpy"]
-
 - repo: https://github.com/b8raoult/optional-dependencies-all
   rev: "0.0.2"
   hooks:
   - id: optional-dependencies-all
-    args: ["--inplace", "--all-key", "all", "--exclude-keys", "dev,docs"]
-
+    args: ["--inplace", "--all-key", "all", "--exclude-keys", "dev,docs,test"]
 - repo: https://github.com/tox-dev/pyproject-fmt
   rev: "2.1.3"
   hooks:
-    - id: pyproject-fmt
+  - id: pyproject-fmt

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anemoi-datasets
-Version: 0.3.3
+Version: 0.3.5
 Summary: A package to hold various functions to support training of ML models on ECMWF data.
 Author-email: "European Centre for Medium-Range Weather Forecasts (ECMWF)" <software.support@ecmwf.int>
 License:                                  Apache License
@@ -238,6 +238,7 @@ Requires-Dist: ecmwflibs>=0.6.3; extra == "all"
 Requires-Dist: entrypoints; extra == "all"
 Requires-Dist: numpy; extra == "all"
 Requires-Dist: pyproj; extra == "all"
+Requires-Dist: pytest; extra == "all"
 Requires-Dist: pyyaml; extra == "all"
 Requires-Dist: requests; extra == "all"
 Requires-Dist: s3fs; extra == "all"
@@ -272,3 +273,5 @@ Provides-Extra: remote
 Requires-Dist: boto3; extra == "remote"
 Requires-Dist: requests; extra == "remote"
 Requires-Dist: s3fs; extra == "remote"
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/docs/building/sources/forcings.yaml

@@ -5,7 +5,7 @@ input:
       args2: value2
     - forcings:
       template: ${input.join.0.source1}
-      params:
+      param:
         - insolation
         - cos_julian_day
         - sin_julian_day

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/pyproject.toml

@@ -67,6 +67,7 @@ optional-dependencies.all = [
   "entrypoints",
   "numpy",
   "pyproj",
+  "pytest",
   "pyyaml",
   "requests",
   "s3fs",
@@ -81,6 +82,7 @@ optional-dependencies.create = [
   "entrypoints",
   "pyproj",
 ]
+
 optional-dependencies.dev = [
   "boto3",
 
@@ -107,6 +109,9 @@ optional-dependencies.remote = [
   "requests",
   "s3fs",
 ]
+optional-dependencies.test = [
+  "pytest",
+]
 urls.Documentation = "https://anemoi-datasets.readthedocs.io/"
 urls.Homepage = "https://github.com/ecmwf/anemoi-datasets/"
 urls.Issues = "https://github.com/ecmwf/anemoi-datasets/issues"

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.3'
-__version_tuple__ = version_tuple = (0, 3, 3)
+__version__ = version = '0.3.5'
+__version_tuple__ = version_tuple = (0, 3, 5)

anemoi_datasets-0.3.3/src/anemoi/datasets/commands/inspect/zarr.py → anemoi_datasets-0.3.5/src/anemoi/datasets/commands/inspect.py

@@ -24,6 +24,9 @@ from anemoi.utils.text import table
 
 from anemoi.datasets import open_dataset
 from anemoi.datasets.data.stores import open_zarr
+from anemoi.datasets.data.stores import zarr_lookup
+
+from . import Command
 
 LOG = logging.getLogger(__name__)
 
@@ -78,46 +81,12 @@ class Version:
         self.metadata = metadata
         self.version = version
         self.dataset = None
-        # try:
         self.dataset = open_dataset(self.path)
-        # except Exception as e:
-        #     LOG.error("Error opening dataset '%s': %s", self.path, e)
 
     def describe(self):
         print(f"📦 Path          : {self.path}")
         print(f"🔢 Format version: {self.version}")
 
-    def probe(self):
-        if "cos_local_time" not in self.name_to_index:
-            print("⚠️ probe: no cos_local_time")
-            return
-
-        try:
-            lon = self.longitudes
-        except AttributeError:
-            print("⚠️ probe: no longitudes")
-            return
-        # print(json.dumps(self.metadata, indent=4))
-        cos_local_time = self.name_to_index["cos_local_time"]
-        data = self.data
-        start, end, frequency = self.first_date, self.last_date, self.frequency
-        date = start
-        same = 0
-        for i in range(10):
-            field = data[i, cos_local_time]
-            buggy = cos_local_time_bug(lon, date).reshape(field.hape)
-            diff = np.abs(field - buggy)
-            if np.max(diff) < 1e-5:
-                same += 1
-            date += datetime.timedelta(hours=frequency)
-            if date > end:
-                break
-        if same > 1:
-            print("❌ probe: cos_local_time is buggy")
-            return
-
-        print("✅ probe: cos_local_time is fixed")
-
     @property
     def name_to_index(self):
         return find(self.metadata, "name_to_index")
@@ -586,29 +555,35 @@ VERSIONS = {
 }
 
 
-class InspectZarr:
-    """Inspect a checkpoint or zarr file."""
+class InspectZarr(Command):
+    """Inspect a zarr dataset."""
 
-    def inspect_zarr(self, path, **kwargs):
-        version = self._info(path)
+    def add_arguments(self, command_parser):
+        command_parser.add_argument("path", metavar="DATASET")
+        command_parser.add_argument("--detailed", action="store_true")
+
+        command_parser.add_argument("--progress", action="store_true")
+        command_parser.add_argument("--statistics", action="store_true")
+        command_parser.add_argument("--size", action="store_true", help="Print size")
 
-        # try:
-        #     with open("/tmp/probe.json", "w") as f:
-        #         json.dump(version.metadata, f, indent=4, sort_keys=True)
-        # except Exception:
-        #     pass
+    def run(self, args):
+        self.inspect_zarr(**vars(args))
+
+    def inspect_zarr(self, path, progress=False, statistics=False, detailed=False, size=False, **kwargs):
+        version = self._info(path)
 
         dotted_line()
         version.describe()
 
         try:
-            if kwargs.get("probe"):
-                return version.probe()
-            if kwargs.get("progress"):
+            if progress:
                 return version.progress()
-            if kwargs.get("statistics"):
+
+            if statistics:
                 return version.brute_force_statistics()
-            version.info(kwargs.get("detailed"), kwargs.get("size"))
+
+            version.info(detailed, size)
+
         except Exception as e:
             LOG.error("Error inspecting zarr file '%s': %s", path, e)
 
@@ -616,14 +591,7 @@ class InspectZarr:
             raise
 
     def _info(self, path):
-        if path.endswith("/"):
-            path = path[:-1]
-
-        try:
-            z = open_zarr(path)
-        except Exception as e:
-            LOG.error("Error opening zarr file '%s': %s", path, e)
-            raise
+        z = open_zarr(zarr_lookup(path))
 
         metadata = dict(z.attrs)
         version = metadata.get("version", "0.0.0")
@@ -640,3 +608,6 @@ class InspectZarr:
                 candidate = klass
 
         return candidate(path, z, metadata, version)
+
+
+command = InspectZarr

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/create/functions/sources/grib.py

@@ -8,6 +8,8 @@
 #
 
 
+import glob
+
 from climetlab import load_source
 from climetlab.utils.patterns import Pattern
 
@@ -22,6 +24,12 @@ def check(ds, paths, **kwargs):
         raise ValueError(f"Expected {count} fields, got {len(ds)} (kwargs={kwargs}, paths={paths})")
 
 
+def _expand(paths):
+    for path in paths:
+        for p in glob.glob(path):
+            yield p
+
+
 def execute(context, dates, path, *args, **kwargs):
     given_paths = path if isinstance(path, list) else [path]
 
@@ -35,7 +43,7 @@ def execute(context, dates, path, *args, **kwargs):
             if name in kwargs:
                 raise ValueError(f"MARS interpolation parameter '{name}' not supported")
 
-        for path in paths:
+        for path in _expand(paths):
             context.trace("📁", "PATH", path)
             s = load_source("file", path)
             s = s.sel(valid_datetime=dates, **kwargs)

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/create/functions/sources/netcdf.py

@@ -7,10 +7,29 @@
 # nor does it submit to any jurisdiction.
 #
 
+import glob
+
 from climetlab import load_source
 from climetlab.utils.patterns import Pattern
 
 
+def _expand(paths):
+    for path in paths:
+        if path.startswith("file://"):
+            path = path[7:]
+
+        if path.startswith("http://"):
+            yield path
+            continue
+
+        if path.startswith("https://"):
+            yield path
+            continue
+
+        for p in glob.glob(path):
+            yield p
+
+
 def check(what, ds, paths, **kwargs):
     count = 1
     for k, v in kwargs.items():
@@ -32,7 +51,7 @@ def load_netcdfs(emoji, what, context, dates, path, *args, **kwargs):
 
         levels = kwargs.get("level", kwargs.get("levelist"))
 
-        for path in paths:
+        for path in _expand(paths):
             context.trace(emoji, what.upper(), path)
             s = load_source("opendap", path)
             s = s.sel(

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/concat.py

@@ -12,7 +12,7 @@ import numpy as np
 
 from .debug import Node
 from .debug import debug_indexing
-from .forewards import Combined
+from .forwards import Combined
 from .indexing import apply_index_to_slices_changes
 from .indexing import expand_list_indexing
 from .indexing import index_to_slices

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/ensemble.py

@@ -8,7 +8,7 @@
 import logging
 
 from .debug import Node
-from .forewards import GivenAxis
+from .forwards import GivenAxis
 from .misc import _auto_adjust
 from .misc import _open
 

anemoi_datasets-0.3.3/src/anemoi/datasets/data/forewards.py → anemoi_datasets-0.3.5/src/anemoi/datasets/data/forwards.py

@@ -91,12 +91,18 @@ class Forwards(Dataset):
     def metadata_specific(self, **kwargs):
         return super().metadata_specific(
             forward=self.forward.metadata_specific(),
+            **self.subclass_metadata_specific(),
             **kwargs,
         )
 
     def source(self, index):
         return self.forward.source(index)
 
+    def subclass_metadata_specific(self):
+        raise NotImplementedError(
+            f"subclass_metadata_specific() must be implemented in derived class {self.__class__.__name__}"
+        )
+
 
 class Combined(Forwards):
     def __init__(self, datasets):

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/grids.py

@@ -12,8 +12,8 @@ import numpy as np
 
 from .debug import Node
 from .debug import debug_indexing
-from .forewards import Combined
-from .forewards import GivenAxis
+from .forwards import Combined
+from .forwards import GivenAxis
 from .indexing import apply_index_to_slices_changes
 from .indexing import expand_list_indexing
 from .indexing import index_to_slices
@@ -91,7 +91,7 @@ class Concat(Combined):
         return Node(self, [d.tree() for d in self.datasets])
 
 
-class Grids(GivenAxis):
+class GridsBase(GivenAxis):
     def __init__(self, datasets, axis):
         super().__init__(datasets, axis)
         # Shape: (dates, variables, ensemble, 1d-values)
@@ -106,7 +106,7 @@ class Grids(GivenAxis):
         pass
 
 
-class ConcatGrids(Grids):
+class Grids(GridsBase):
     # TODO: select the statistics of the most global grid?
     @property
     def latitudes(self):
@@ -127,7 +127,7 @@ class ConcatGrids(Grids):
         return Node(self, [d.tree() for d in self.datasets], mode="concat")
 
 
-class Cutout(Grids):
+class Cutout(GridsBase):
     def __init__(self, datasets, axis):
         from anemoi.datasets.grids import cutout_mask
 
@@ -220,7 +220,7 @@ def grids_factory(args, kwargs):
     datasets = [_open(e) for e in grids]
     datasets, kwargs = _auto_adjust(datasets, kwargs)
 
-    return ConcatGrids(datasets, axis=axis)._subset(**kwargs)
+    return Grids(datasets, axis=axis)._subset(**kwargs)
 
 
 def cutout_factory(args, kwargs):

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/join.py

@@ -13,7 +13,7 @@ import numpy as np
 from .debug import Node
 from .debug import Source
 from .debug import debug_indexing
-from .forewards import Combined
+from .forwards import Combined
 from .indexing import apply_index_to_slices_changes
 from .indexing import expand_list_indexing
 from .indexing import index_to_slices

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/masked.py

@@ -14,7 +14,7 @@ from ..grids import cropping_mask
 from .dataset import Dataset
 from .debug import Node
 from .debug import debug_indexing
-from .forewards import Forwards
+from .forwards import Forwards
 from .indexing import apply_index_to_slices_changes
 from .indexing import expand_list_indexing
 from .indexing import index_to_slices
@@ -83,6 +83,9 @@ class Thinning(Masked):
     def tree(self):
         return Node(self, [self.forward.tree()], thinning=self.thinning, method=self.method)
 
+    def subclass_metadata_specific(self):
+        return dict(thinning=self.thinning, method=self.method)
+
 
 class Cropping(Masked):
     def __init__(self, forward, area):
@@ -104,3 +107,6 @@ class Cropping(Masked):
 
     def tree(self):
         return Node(self, [self.forward.tree()], area=self.area)
+
+    def metadata_specific(self, **kwargs):
+        return super().metadata_specific(area=self.area, **kwargs)

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/select.py

@@ -11,7 +11,7 @@ from functools import cached_property
 from .debug import Node
 from .debug import Source
 from .debug import debug_indexing
-from .forewards import Forwards
+from .forwards import Forwards
 from .indexing import apply_index_to_slices_changes
 from .indexing import expand_list_indexing
 from .indexing import index_to_slices
@@ -88,6 +88,10 @@ class Select(Forwards):
     def tree(self):
         return Node(self, [self.dataset.tree()], **self.title)
 
+    def subclass_metadata_specific(self):
+        # return dict(indices=self.indices)
+        return {}
+
 
 class Rename(Forwards):
     def __init__(self, dataset, rename):
@@ -105,8 +109,8 @@ class Rename(Forwards):
     def name_to_index(self):
         return {k: i for i, k in enumerate(self.variables)}
 
-    def metadata_specific(self, **kwargs):
-        return super().metadata_specific(rename=self.rename, **kwargs)
-
     def tree(self):
         return Node(self, [self.forward.tree()], rename=self.rename)
+
+    def subclass_metadata_specific(self):
+        return dict(rename=self.rename)

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/statistics.py

@@ -10,7 +10,7 @@ from functools import cached_property
 
 from . import open_dataset
 from .debug import Node
-from .forewards import Forwards
+from .forwards import Forwards
 
 LOG = logging.getLogger(__name__)
 
@@ -34,11 +34,8 @@ class Statistics(Forwards):
             delta = self.frequency
         return self._statistic.statistics_tendencies(delta)
 
-    def metadata_specific(self, **kwargs):
-        return super().metadata_specific(
-            statistics=self._statistic.metadata_specific(),
-            **kwargs,
-        )
+    def subclass_metadata_specific(self):
+        return dict(statistics=self._statistic.metadata_specific())
 
     def tree(self):
         return Node(self, [self.forward.tree()])

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/stores.py

@@ -347,7 +347,7 @@ class ZarrWithMissingDates(Zarr):
         return "zarr*"
 
 
-def zarr_lookup(name):
+def zarr_lookup(name, fail=True):
 
     if name.endswith(".zarr") or name.endswith(".zip"):
         return name
@@ -372,4 +372,7 @@ def zarr_lookup(name):
         except zarr.errors.PathNotFoundError:
             pass
 
-    raise ValueError(f"Cannot find a dataset that matched '{name}'. Tried: {tried}")
+    if fail:
+        raise ValueError(f"Cannot find a dataset that matched '{name}'. Tried: {tried}")
+
+    return None

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/subset.py

@@ -13,7 +13,7 @@ import numpy as np
 from .debug import Node
 from .debug import Source
 from .debug import debug_indexing
-from .forewards import Forwards
+from .forwards import Forwards
 from .indexing import apply_index_to_slices_changes
 from .indexing import expand_list_indexing
 from .indexing import index_to_slices
@@ -109,3 +109,9 @@ class Subset(Forwards):
 
     def tree(self):
         return Node(self, [self.dataset.tree()], **self.reason)
+
+    def subclass_metadata_specific(self):
+        return {
+            # "indices": self.indices,
+            "reason": self.reason,
+        }

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/data/unchecked.py

@@ -11,7 +11,7 @@ from functools import wraps
 
 from .concat import ConcatMixin
 from .debug import Node
-from .forewards import Combined
+from .forwards import Combined
 from .misc import _auto_adjust
 from .misc import _open
 

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi/datasets/dates/__init__.py

@@ -116,8 +116,8 @@ class StartEndDates(Dates):
         start = no_time_zone(start)
         end = no_time_zone(end)
 
-        if end <= start:
-            raise ValueError(f"End date {end} must be after start date {start}")
+        # if end <= start:
+        #     raise ValueError(f"End date {end} must be after start date {start}")
 
         increment = datetime.timedelta(hours=frequency)
 

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi_datasets.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anemoi-datasets
-Version: 0.3.3
+Version: 0.3.5
 Summary: A package to hold various functions to support training of ML models on ECMWF data.
 Author-email: "European Centre for Medium-Range Weather Forecasts (ECMWF)" <software.support@ecmwf.int>
 License:                                  Apache License
@@ -238,6 +238,7 @@ Requires-Dist: ecmwflibs>=0.6.3; extra == "all"
 Requires-Dist: entrypoints; extra == "all"
 Requires-Dist: numpy; extra == "all"
 Requires-Dist: pyproj; extra == "all"
+Requires-Dist: pytest; extra == "all"
 Requires-Dist: pyyaml; extra == "all"
 Requires-Dist: requests; extra == "all"
 Requires-Dist: s3fs; extra == "all"
@@ -272,3 +273,5 @@ Provides-Extra: remote
 Requires-Dist: boto3; extra == "remote"
 Requires-Dist: requests; extra == "remote"
 Requires-Dist: s3fs; extra == "remote"
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi_datasets.egg-info/SOURCES.txt

@@ -157,9 +157,8 @@ src/anemoi/datasets/commands/__init__.py
 src/anemoi/datasets/commands/compare.py
 src/anemoi/datasets/commands/copy.py
 src/anemoi/datasets/commands/create.py
+src/anemoi/datasets/commands/inspect.py
 src/anemoi/datasets/commands/scan.py
-src/anemoi/datasets/commands/inspect/__init__.py
-src/anemoi/datasets/commands/inspect/zarr.py
 src/anemoi/datasets/compute/__init__.py
 src/anemoi/datasets/compute/recentre.py
 src/anemoi/datasets/create/__init__.py
@@ -203,7 +202,7 @@ src/anemoi/datasets/data/dataset.py
 src/anemoi/datasets/data/debug.css
 src/anemoi/datasets/data/debug.py
 src/anemoi/datasets/data/ensemble.py
-src/anemoi/datasets/data/forewards.py
+src/anemoi/datasets/data/forwards.py
 src/anemoi/datasets/data/grids.py
 src/anemoi/datasets/data/indexing.py
 src/anemoi/datasets/data/join.py
@@ -225,7 +224,6 @@ src/anemoi_datasets.egg-info/requires.txt
 src/anemoi_datasets.egg-info/top_level.txt
 tests/create-perturbations-full.yaml
 tests/create-shift.yaml
-tests/requirements.txt
 tests/test_chunks.py
 tests/test_data.py
 tests/test_dates.py

{anemoi_datasets-0.3.3 → anemoi_datasets-0.3.5}/src/anemoi_datasets.egg-info/requires.txt

@@ -14,6 +14,7 @@ ecmwflibs>=0.6.3
 entrypoints
 numpy
 pyproj
+pytest
 pyyaml
 requests
 s3fs
@@ -52,3 +53,6 @@ sphinx-rtd-theme
 boto3
 requests
 s3fs
+
+[test]
+pytest