Fix running unit tests locally

.gitignore

@@ -175,3 +175,5 @@ fair-mast-*.e*
 fair-mast-*.o*
 .s5cfg*
 *.db
+
+uv.lock

pyproject.toml

@@ -25,6 +25,7 @@ dependencies = {file = ["requirements.txt"]}
 dev = [
     "pytest",
     "pytest-mock",
+    "pytest-env",
     "ruff"
 ]
 
@@ -36,4 +37,9 @@ mpi = [
 [project.urls]
 
 [tool.ruff]
-exclude = [".venv", "notebooks"]
+exclude = [".venv", "notebooks"]
+
+[tool.pytest_env]
+UDA_HOST = "uda2.mast.l"
+UDA_META_PLUGINNAME = "MASTU_DB"
+UDA_METANEW_PLUGINNAME = "MAST_DB"

src/core/writer.py

@@ -52,10 +52,10 @@ def _convert_dict_attrs_to_json(self, dataset: xr.Dataset):
 
 class ZarrDatasetWriter(DatasetWriter):
     def __init__(
-        self, output_path: str, mode: str = "single", zarr_format: int = 2, **kwargs
+        self, output_path: str, mode: str = "single", zarr_version: int = 2, **kwargs
     ):
         super().__init__(output_path)
-        self.version = zarr_format
+        self.version = zarr_version
         self.mode = mode
 
     @property
@@ -73,15 +73,19 @@ def write(self, file_name: str, group_name: str, dataset: xr.Dataset):
     def _write_single_zarr(self, file_name: str, name: str, dataset: xr.Dataset):
         file_name = self.output_path / file_name
         dataset.to_zarr(
-            file_name, group=name, mode="w", zarr_format=self.version, consolidated=True
+            file_name,
+            group=name,
+            mode="w",
+            zarr_version=self.version,
+            consolidated=True,
         )
         zarr.consolidate_metadata(file_name)
 
     def _write_multi_zarr(self, file_name: str, name: str, dataset: xr.Dataset):
         file_name = Path(file_name)
         path = self.output_path / f"{file_name.stem}/{name}.zarr"
         path.parent.mkdir(exist_ok=True, parents=True)
-        dataset.to_zarr(path, mode="a", zarr_format=self.version, consolidated=True)
+        dataset.to_zarr(path, mode="a", zarr_version=self.version, consolidated=True)
         zarr.consolidate_metadata(path)
 
 

tests/core/test_loader.py

@@ -6,7 +6,18 @@
 from src.core.load import SALLoader, UDALoader, ZarrLoader
 
 
-@pytest.mark.skip(reason="Pyuda client unavailable")
+def try_uda():
+    try:
+        import pyuda
+
+        client = pyuda.Client()
+        client.get("ip", 30421)
+        return False
+    except Exception:
+        return True
+
+
+@pytest.mark.skipif(try_uda(), reason="Pyuda client unavailable")
 def test_load_uda():
     loader = UDALoader()
     signal = loader.load(30420, "ip")