From 85647467546046b1e6a0295bd0a9aa6870760a60 Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 10:45:30 -0700
Subject: [PATCH 01/11] Change NowcastWorker mock to pytest fixture

Test suite maintenance.

re: issue #81
---
 nowcast/workers/make_CHS_currents_file.py    |  5 +-
 tests/workers/test_make_CHS_currents_file.py | 68 +++++++++-----------
 2 files changed, 33 insertions(+), 40 deletions(-)

diff --git a/nowcast/workers/make_CHS_currents_file.py b/nowcast/workers/make_CHS_currents_file.py
index ef0716b8..df16dcb2 100644
--- a/nowcast/workers/make_CHS_currents_file.py
+++ b/nowcast/workers/make_CHS_currents_file.py
@@ -12,8 +12,8 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
-"""SalishSeaCast worker that average, unstaggers and rotates the near
-surface velocities, and writes them out in an nc file for CHS to use
+"""SalishSeaCast worker that average, unstaggers and rotates the near surface velocities,
+and writes them out in an nc file for CHS to use
 """
 import logging
 import os
@@ -53,6 +53,7 @@ def main():
         help="Date to process the velocities for.",
     )
     worker.run(make_CHS_currents_file, success, failure)
+    return worker
 
 
 def success(parsed_args):
diff --git a/tests/workers/test_make_CHS_currents_file.py b/tests/workers/test_make_CHS_currents_file.py
index 01d4433c..1d6b6bbc 100644
--- a/tests/workers/test_make_CHS_currents_file.py
+++ b/tests/workers/test_make_CHS_currents_file.py
@@ -21,7 +21,7 @@
 import textwrap
 from pathlib import Path
 from types import SimpleNamespace
-from unittest.mock import Mock, patch
+from unittest.mock import patch
 
 import arrow
 import nemo_nowcast
@@ -64,48 +64,40 @@ def config(base_config):
     return config_
 
 
-@patch("nowcast.workers.make_CHS_currents_file.NowcastWorker", spec=True)
+@pytest.fixture
+def mock_worker(mock_nowcast_worker, monkeypatch):
+    monkeypatch.setattr(make_CHS_currents_file, "NowcastWorker", mock_nowcast_worker)
+
+
 class TestMain:
     """Unit tests for main() function."""
 
-    def test_instantiate_worker(self, m_worker):
-        m_worker().cli = Mock(name="cli")
-        make_CHS_currents_file.main()
-        args, kwargs = m_worker.call_args
-        assert args == ("make_CHS_currents_file",)
-        assert list(kwargs.keys()) == ["description"]
-
-    def test_init_cli(self, m_worker):
-        m_worker().cli = Mock(name="cli")
-        make_CHS_currents_file.main()
-        m_worker().init_cli.assert_called_once_with()
-
-    def test_add_run_type_arg(self, m_worker):
-        m_worker().cli = Mock(name="cli")
-        make_CHS_currents_file.main()
-        args, kwargs = m_worker().cli.add_argument.call_args_list[0]
-        assert args == ("run_type",)
-        assert kwargs["choices"] == {"nowcast", "forecast", "forecast2"}
-        assert "help" in kwargs
-
-    def test_add_run_date_option(self, m_worker):
-        m_worker().cli = Mock(name="cli")
-        make_CHS_currents_file.main()
-        args, kwargs = m_worker().cli.add_date_option.call_args_list[0]
-        assert args == ("--run-date",)
-        assert kwargs["default"] == arrow.now().floor("day")
-        assert "help" in kwargs
-
-    def test_run_worker(self, m_worker):
-        m_worker().cli = Mock(name="cli")
-        make_CHS_currents_file.main()
-        args, kwargs = m_worker().run.call_args
-        assert args == (
-            make_CHS_currents_file.make_CHS_currents_file,
-            make_CHS_currents_file.success,
-            make_CHS_currents_file.failure,
+    def test_instantiate_worker(self, mock_worker):
+        worker = make_CHS_currents_file.main()
+        assert worker.name == "make_CHS_currents_file"
+        assert worker.description.startswith(
+            "SalishSeaCast worker that average, unstaggers and rotates the near surface velocities,"
         )
 
+    def test_add_run_type_arg(self, mock_worker):
+        worker = make_CHS_currents_file.main()
+        assert worker.cli.parser._actions[3].dest == "run_type"
+        assert worker.cli.parser._actions[3].choices == {
+            "nowcast",
+            "forecast",
+            "forecast2",
+        }
+        assert worker.cli.parser._actions[3].help
+
+    def test_add_run_date_option(self, mock_worker):
+        worker = make_CHS_currents_file.main()
+        assert worker.cli.parser._actions[4].dest == "run_date"
+        expected = nemo_nowcast.cli.CommandLineInterface.arrow_date
+        assert worker.cli.parser._actions[4].type == expected
+        expected = arrow.now().floor("day")
+        assert worker.cli.parser._actions[4].default == expected
+        assert worker.cli.parser._actions[4].help
+
 
 class TestConfig:
     """Unit tests for production YAML config file elements related to worker."""

From 2a2d15deda047b09b7f9fd6825c3b74d1e566054 Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 10:57:30 -0700
Subject: [PATCH 02/11] Refactor run_type conditional to match case

Updated the run_type conditionals from if-elif to a match case statement for
better clarity and future extensibility. Added error handling to raise a
WorkerError if the run_type is unexpected. Although the CLI parser handles that
error, adding default case handling prevents static analysis warnings about
possible variables references before assignment.
---
 nowcast/workers/make_CHS_currents_file.py | 22 +++++++++++++---------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/nowcast/workers/make_CHS_currents_file.py b/nowcast/workers/make_CHS_currents_file.py
index df16dcb2..79e89573 100644
--- a/nowcast/workers/make_CHS_currents_file.py
+++ b/nowcast/workers/make_CHS_currents_file.py
@@ -21,6 +21,7 @@
 
 import arrow
 import xarray
+from build.lib.nemo_nowcast.worker import WorkerError
 from nemo_nowcast import NowcastWorker
 from salishsea_tools import viz_tools
 
@@ -98,15 +99,18 @@ def make_CHS_currents_file(parsed_args, config, *args):
     """
     run_type = parsed_args.run_type
     run_date = parsed_args.run_date
-    if run_type == "nowcast":
-        start_date = run_date.format("YYYYMMDD")
-        end_date = run_date.format("YYYYMMDD")
-    elif run_type == "forecast":
-        start_date = run_date.shift(days=1).format("YYYYMMDD")
-        end_date = run_date.shift(days=2).format("YYYYMMDD")
-    elif run_type == "forecast2":
-        start_date = run_date.shift(days=2).format("YYYYMMDD")
-        end_date = run_date.shift(days=3).format("YYYYMMDD")
+    match run_type:
+        case "nowcast":
+            start_date = run_date.format("YYYYMMDD")
+            end_date = run_date.format("YYYYMMDD")
+        case "forecast":
+            start_date = run_date.shift(days=1).format("YYYYMMDD")
+            end_date = run_date.shift(days=2).format("YYYYMMDD")
+        case "forecast2":
+            start_date = run_date.shift(days=2).format("YYYYMMDD")
+            end_date = run_date.shift(days=3).format("YYYYMMDD")
+        case _:
+            raise WorkerError(f"unexpected run type: {run_type}")
 
     grid_dir = Path(config["figures"]["grid dir"])
     meshfilename = grid_dir / config["run types"][run_type]["mesh mask"]

From c79afb8cf785f48608926441f77f112595451388 Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 10:59:44 -0700
Subject: [PATCH 03/11] Fix param type definitions in docstrings

Corrected the format of parameter type definitions in the docstrings for
consistency and clarity. This change ensures better readability and proper
documentation standards.
---
 nowcast/workers/make_CHS_currents_file.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/nowcast/workers/make_CHS_currents_file.py b/nowcast/workers/make_CHS_currents_file.py
index 79e89573..70f362db 100644
--- a/nowcast/workers/make_CHS_currents_file.py
+++ b/nowcast/workers/make_CHS_currents_file.py
@@ -144,9 +144,9 @@ def _read_avg_unstagger_rotate(meshfilename, src_dir, ufile, vfile, run_type):
     """
     :param str meshfilename:
     :param :py:class:`pathlib.Path` src_dir:
-    :param str: ufile:
-    :param str: vfile:
-    :param str: run_type:
+    :param str ufile:
+    :param str vfile:
+    :param str run_type:
 
     :return: 4_tuple of data arrays
                urot5: east velocity averaged over top 5 grid cells
@@ -203,7 +203,7 @@ def _write_netcdf(src_dir, urot5, vrot5, urot10, vrot10, run_type):
     :param :py:class:`xarray.DataArray` vrot5:
     :param :py:class:`xarray.DataArray` urot10:
     :param :py:class:`xarray.DataArray` vrot10:
-    :param str: run_type:
+    :param str run_type:
 
     :return: str CHS_currents_filename
     """

From 5613d9872c6042d6f433bbdaf8b765131b941732 Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 11:29:41 -0700
Subject: [PATCH 04/11] Change logging mocks to pytest caplog fixture

Replace unittest.mock.patch decorator with pytest caplog fixture for tests of
logging.

Test suite maintenance re: issue #82.
---
 tests/workers/test_make_CHS_currents_file.py | 38 +++++++++++++-------
 1 file changed, 25 insertions(+), 13 deletions(-)

diff --git a/tests/workers/test_make_CHS_currents_file.py b/tests/workers/test_make_CHS_currents_file.py
index 1d6b6bbc..ace296ab 100644
--- a/tests/workers/test_make_CHS_currents_file.py
+++ b/tests/workers/test_make_CHS_currents_file.py
@@ -18,6 +18,7 @@
 
 """Unit tests for SalishSeaCast make_CHS_currents_file worker.
 """
+import logging
 import textwrap
 from pathlib import Path
 from types import SimpleNamespace
@@ -148,30 +149,38 @@ def test_file_group(self, prod_config):
 
 
 @pytest.mark.parametrize("run_type", ["nowcast", "forecast", "forecast2"])
-@patch("nowcast.workers.make_CHS_currents_file.logger", autospec=True)
 class TestSuccess:
     """Unit tests for success() function."""
 
-    def test_success(self, m_logger, run_type):
-        parsed_args = SimpleNamespace(
-            run_type=run_type, run_date=arrow.get("2018-09-01")
-        )
+    def test_success(self, run_type, caplog):
+        run_date = arrow.get("2018-09-01")
+        parsed_args = SimpleNamespace(run_type=run_type, run_date=run_date)
+        caplog.set_level(logging.DEBUG)
+
         msg_type = make_CHS_currents_file.success(parsed_args)
-        assert m_logger.info.called
+
+        assert caplog.records[0].levelname == "INFO"
+        expected = (
+            f"Made CHS currents file for {run_date.format("YYYY-MM-DD")} for {run_type}"
+        )
+        assert caplog.records[0].message == expected
         assert msg_type == f"success {run_type}"
 
 
 @pytest.mark.parametrize("run_type", ["nowcast", "forecast", "forecast2"])
-@patch("nowcast.workers.make_CHS_currents_file.logger", autospec=True)
 class TestFailure:
     """Unit tests for failure() function."""
 
-    def test_failure(self, m_logger, run_type):
-        parsed_args = SimpleNamespace(
-            run_type=run_type, run_date=arrow.get("2018-09-01")
-        )
+    def test_failure(self, run_type, caplog):
+        run_date = arrow.get("2018-09-01")
+        parsed_args = SimpleNamespace(run_type=run_type, run_date=run_date)
+        caplog.set_level(logging.DEBUG)
+
         msg_type = make_CHS_currents_file.failure(parsed_args)
-        assert m_logger.critical.called
+
+        assert caplog.records[0].levelname == "CRITICAL"
+        expected = f"Making CHS currents file for {run_date.format("YYYY-MM-DD")} failed for {run_type}"
+        assert caplog.records[0].message == expected
         assert msg_type == f"failure {run_type}"
 
 
@@ -189,7 +198,9 @@ class TestMakeCHSCurrentsFile:
     """Unit tests for make_CHS_currents_function."""
 
     @pytest.mark.parametrize("run_type", ["nowcast", "forecast", "forecast2"])
-    def test_checklist(self, m_fix_perms, m_write_ncdf, m_read_aur, run_type, config):
+    def test_checklist(
+        self, m_fix_perms, m_write_ncdf, m_read_aur, run_type, config, caplog
+    ):
         parsed_args = SimpleNamespace(
             run_type=run_type, run_date=arrow.get("2018-09-01")
         )
@@ -230,6 +241,7 @@ def test_read_avg_unstagger_rotatehecklist(
         ufile,
         vfile,
         config,
+        caplog,
     ):
         parsed_args = SimpleNamespace(
             run_type=run_type, run_date=arrow.get("2018-09-01")

From 0dad5fa0292de52e55cf584da7217de0913252e0 Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 11:40:46 -0700
Subject: [PATCH 05/11] Refactor lib.fix_perms patch to pytest fixture

Replaced direct patching of fix_perms with a pytest fixture for better test
structure and readability. This ensures the mock is applied more cleanly and
isolated within the test context.
---
 tests/workers/test_make_CHS_currents_file.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/tests/workers/test_make_CHS_currents_file.py b/tests/workers/test_make_CHS_currents_file.py
index ace296ab..5985db30 100644
--- a/tests/workers/test_make_CHS_currents_file.py
+++ b/tests/workers/test_make_CHS_currents_file.py
@@ -193,13 +193,20 @@ def test_failure(self, run_type, caplog):
     "nowcast.workers.make_CHS_currents_file._write_netcdf",
     spec=make_CHS_currents_file._write_netcdf,
 )
-@patch("nowcast.workers.make_CHS_currents_file.lib.fix_perms", autospec=True)
 class TestMakeCHSCurrentsFile:
     """Unit tests for make_CHS_currents_function."""
 
+    @staticmethod
+    @pytest.fixture
+    def mock_fix_perms(monkeypatch):
+        def _mock_fix_perms(path, grp_name):
+            pass
+
+        monkeypatch.setattr(make_CHS_currents_file.lib, "fix_perms", _mock_fix_perms)
+
     @pytest.mark.parametrize("run_type", ["nowcast", "forecast", "forecast2"])
     def test_checklist(
-        self, m_fix_perms, m_write_ncdf, m_read_aur, run_type, config, caplog
+        self, m_write_ncdf, m_read_aur, mock_fix_perms, run_type, config, caplog
     ):
         parsed_args = SimpleNamespace(
             run_type=run_type, run_date=arrow.get("2018-09-01")
@@ -233,9 +240,9 @@ def test_checklist(
     )
     def test_read_avg_unstagger_rotatehecklist(
         self,
-        m_fix_perms,
         m_write_ncdf,
         m_read_aur,
+        mock_fix_perms,
         run_type,
         results_archive,
         ufile,

From 1ec9064dab15853d7b7220071b8c5f6645075d04 Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 13:39:05 -0700
Subject: [PATCH 06/11] Refactor `_write_netcdf` patch to pytest fixture

Replaced direct patching with a pytest fixture to mock the `_write_netcdf`
function in unit tests for `make_CHS_currents_file`. This improves test
maintainability and readability. Reorganized parameterized tests to adapt to the
new mocking approach.
---
 tests/workers/test_make_CHS_currents_file.py | 26 ++++++++++++--------
 1 file changed, 16 insertions(+), 10 deletions(-)

diff --git a/tests/workers/test_make_CHS_currents_file.py b/tests/workers/test_make_CHS_currents_file.py
index 5985db30..dbf796a2 100644
--- a/tests/workers/test_make_CHS_currents_file.py
+++ b/tests/workers/test_make_CHS_currents_file.py
@@ -189,13 +189,17 @@ def test_failure(self, run_type, caplog):
     return_value=("urot5", "vrot5", "urot10", "vrot10"),
     autospec=True,
 )
-@patch(
-    "nowcast.workers.make_CHS_currents_file._write_netcdf",
-    spec=make_CHS_currents_file._write_netcdf,
-)
 class TestMakeCHSCurrentsFile:
     """Unit tests for make_CHS_currents_function."""
 
+    @staticmethod
+    @pytest.fixture
+    def mock_write_netcdf(monkeypatch):
+        def _mock_write_netcdf(src_dir, urot5, vrot5, urot10, vrot10, run_type):
+            return f"{src_dir}/CHS_currents.nc"
+
+        monkeypatch.setattr(make_CHS_currents_file, "_write_netcdf", _mock_write_netcdf)
+
     @staticmethod
     @pytest.fixture
     def mock_fix_perms(monkeypatch):
@@ -206,13 +210,15 @@ def _mock_fix_perms(path, grp_name):
 
     @pytest.mark.parametrize("run_type", ["nowcast", "forecast", "forecast2"])
     def test_checklist(
-        self, m_write_ncdf, m_read_aur, mock_fix_perms, run_type, config, caplog
+        self, m_read_aur, mock_write_netcdf, mock_fix_perms, run_type, config, caplog
     ):
-        parsed_args = SimpleNamespace(
-            run_type=run_type, run_date=arrow.get("2018-09-01")
-        )
+        run_date = arrow.get("2018-09-01")
+        parsed_args = SimpleNamespace(run_type=run_type, run_date=run_date)
         checklist = make_CHS_currents_file.make_CHS_currents_file(parsed_args, config)
-        expected = {run_type: {"filename": m_write_ncdf(), "run date": "2018-09-01"}}
+        expected_filepath = f"{Path(config["results archive"][run_type])
+            /run_date.format("DDMMMYY").lower()
+            /"CHS_currents.nc"}"
+        expected = {run_type: {"filename": expected_filepath, "run date": "2018-09-01"}}
         assert checklist == expected
 
     @pytest.mark.parametrize(
@@ -240,8 +246,8 @@ def test_checklist(
     )
     def test_read_avg_unstagger_rotatehecklist(
         self,
-        m_write_ncdf,
         m_read_aur,
+        mock_write_netcdf,
         mock_fix_perms,
         run_type,
         results_archive,

From 194d2120179dc40cad67c88f59c635c22ae13f44 Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 13:48:08 -0700
Subject: [PATCH 07/11] Refactor `_read_avg_unstagger_rotate` patch to pytest
 fixture

Replace the @patch decorator with a pytest fixture for
mock_read_avg_unstagger_rotate. This approach provides better clarity and
modularity in the test structure, making it easier to manage and extend the
tests. Removed the outdated test method that relied on the @patch decorator.
---
 tests/workers/test_make_CHS_currents_file.py | 75 ++++++--------------
 1 file changed, 21 insertions(+), 54 deletions(-)

diff --git a/tests/workers/test_make_CHS_currents_file.py b/tests/workers/test_make_CHS_currents_file.py
index dbf796a2..70ee5633 100644
--- a/tests/workers/test_make_CHS_currents_file.py
+++ b/tests/workers/test_make_CHS_currents_file.py
@@ -22,7 +22,6 @@
 import textwrap
 from pathlib import Path
 from types import SimpleNamespace
-from unittest.mock import patch
 
 import arrow
 import nemo_nowcast
@@ -184,14 +183,23 @@ def test_failure(self, run_type, caplog):
         assert msg_type == f"failure {run_type}"
 
 
-@patch(
-    "nowcast.workers.make_CHS_currents_file._read_avg_unstagger_rotate",
-    return_value=("urot5", "vrot5", "urot10", "vrot10"),
-    autospec=True,
-)
 class TestMakeCHSCurrentsFile:
     """Unit tests for make_CHS_currents_function."""
 
+    @staticmethod
+    @pytest.fixture
+    def mock_read_avg_unstagger_rotate(monkeypatch):
+        def _mock_read_avg_unstagger_rotate(
+            meshfilename, src_dir, ufile, vfile, run_type
+        ):
+            return "urot5", "vrot5", "urot10", "vrot10"
+
+        monkeypatch.setattr(
+            make_CHS_currents_file,
+            "_read_avg_unstagger_rotate",
+            _mock_read_avg_unstagger_rotate,
+        )
+
     @staticmethod
     @pytest.fixture
     def mock_write_netcdf(monkeypatch):
@@ -210,7 +218,13 @@ def _mock_fix_perms(path, grp_name):
 
     @pytest.mark.parametrize("run_type", ["nowcast", "forecast", "forecast2"])
     def test_checklist(
-        self, m_read_aur, mock_write_netcdf, mock_fix_perms, run_type, config, caplog
+        self,
+        mock_read_avg_unstagger_rotate,
+        mock_write_netcdf,
+        mock_fix_perms,
+        run_type,
+        config,
+        caplog,
     ):
         run_date = arrow.get("2018-09-01")
         parsed_args = SimpleNamespace(run_type=run_type, run_date=run_date)
@@ -220,50 +234,3 @@ def test_checklist(
             /"CHS_currents.nc"}"
         expected = {run_type: {"filename": expected_filepath, "run date": "2018-09-01"}}
         assert checklist == expected
-
-    @pytest.mark.parametrize(
-        "run_type, results_archive, ufile, vfile",
-        [
-            (
-                "nowcast",
-                "nowcast-blue",
-                "SalishSea_1h_20180901_20180901_grid_U.nc",
-                "SalishSea_1h_20180901_20180901_grid_V.nc",
-            ),
-            (
-                "forecast",
-                "forecast",
-                "SalishSea_1h_20180902_20180903_grid_U.nc",
-                "SalishSea_1h_20180902_20180903_grid_V.nc",
-            ),
-            (
-                "forecast2",
-                "forecast2",
-                "SalishSea_1h_20180903_20180904_grid_U.nc",
-                "SalishSea_1h_20180903_20180904_grid_V.nc",
-            ),
-        ],
-    )
-    def test_read_avg_unstagger_rotatehecklist(
-        self,
-        m_read_aur,
-        mock_write_netcdf,
-        mock_fix_perms,
-        run_type,
-        results_archive,
-        ufile,
-        vfile,
-        config,
-        caplog,
-    ):
-        parsed_args = SimpleNamespace(
-            run_type=run_type, run_date=arrow.get("2018-09-01")
-        )
-        make_CHS_currents_file.make_CHS_currents_file(parsed_args, config)
-        m_read_aur.assert_called_once_with(
-            Path("nowcast-sys/grid/mesh_mask201702.nc"),
-            Path(results_archive) / "01sep18",
-            ufile,
-            vfile,
-            run_type,
-        )

From d9f74338a75e632ce39ceee5a617bfcb363aedc4 Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 14:13:05 -0700
Subject: [PATCH 08/11] Update dropping variable in make_CHS_currents_file

Replaced the deprecated 'drop' method with 'drop_vars' for removing the
'time_centered' variable. This ensures compatibility with newer versions of the
xarray library.
---
 nowcast/workers/make_CHS_currents_file.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nowcast/workers/make_CHS_currents_file.py b/nowcast/workers/make_CHS_currents_file.py
index 70f362db..fc16a0fb 100644
--- a/nowcast/workers/make_CHS_currents_file.py
+++ b/nowcast/workers/make_CHS_currents_file.py
@@ -247,7 +247,7 @@ def _write_netcdf(src_dir, urot5, vrot5, urot10, vrot10, run_type):
         "units": "m/s",
     }
 
-    myds = myds.drop("time_centered")
+    myds = myds.drop_vars("time_centered")
     myds = myds.rename({"x": "gridX", "y": "gridY"})
 
     encoding = {

From 0681d0f2bdc7e8c8647ba94254a4d1728f9a30aa Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 14:14:26 -0700
Subject: [PATCH 09/11] Correct unlimited dimension in netCDF export

Correct the unlimited dimension from "time" to "time_counter" when exporting
to netCDF in make_CHS_currents_file.py.
---
 nowcast/workers/make_CHS_currents_file.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nowcast/workers/make_CHS_currents_file.py b/nowcast/workers/make_CHS_currents_file.py
index fc16a0fb..551a80db 100644
--- a/nowcast/workers/make_CHS_currents_file.py
+++ b/nowcast/workers/make_CHS_currents_file.py
@@ -287,7 +287,7 @@ def _write_netcdf(src_dir, urot5, vrot5, urot10, vrot10, run_type):
     }
 
     filename = src_dir / "CHS_currents.nc"
-    myds.to_netcdf(filename, encoding=encoding, unlimited_dims=("time",))
+    myds.to_netcdf(filename, encoding=encoding, unlimited_dims=("time_counter",))
 
     logger.debug(f"{run_type}: netcdf file written: {filename}")
 

From 48ad20e0a0690405f120eb96dff40210d47885de Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 14:30:06 -0700
Subject: [PATCH 10/11] Correct import statement for WorkerError

PyCharm generated an import from a build/ directory that was just plain wrong!
---
 nowcast/workers/make_CHS_currents_file.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/nowcast/workers/make_CHS_currents_file.py b/nowcast/workers/make_CHS_currents_file.py
index 551a80db..4896279e 100644
--- a/nowcast/workers/make_CHS_currents_file.py
+++ b/nowcast/workers/make_CHS_currents_file.py
@@ -21,8 +21,7 @@
 
 import arrow
 import xarray
-from build.lib.nemo_nowcast.worker import WorkerError
-from nemo_nowcast import NowcastWorker
+from nemo_nowcast import NowcastWorker, WorkerError
 from salishsea_tools import viz_tools
 
 from nowcast import lib

From d78c51abd35e802920e511577b43706a228419f5 Mon Sep 17 00:00:00 2001
From: Doug Latornell <dlatornell@eoas.ubc.ca>
Date: Mon, 2 Sep 2024 14:33:20 -0700
Subject: [PATCH 11/11] Improve worker module docstring

Corrected grammatical error and improved the wording of module docstring in
`make_CHS_currents_file.py` and its associated tests. Ensured consistent use of
terminology by changing "average" to "averages" and specifying "netCDF" instead
of "nc file".
---
 nowcast/workers/make_CHS_currents_file.py    | 4 ++--
 tests/workers/test_make_CHS_currents_file.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/nowcast/workers/make_CHS_currents_file.py b/nowcast/workers/make_CHS_currents_file.py
index 4896279e..93b58c62 100644
--- a/nowcast/workers/make_CHS_currents_file.py
+++ b/nowcast/workers/make_CHS_currents_file.py
@@ -12,8 +12,8 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
-"""SalishSeaCast worker that average, unstaggers and rotates the near surface velocities,
-and writes them out in an nc file for CHS to use
+"""SalishSeaCast worker that averages, unstaggers and rotates the near surface velocities,
+and writes them out in a netCDF file for CHS to use.
 """
 import logging
 import os
diff --git a/tests/workers/test_make_CHS_currents_file.py b/tests/workers/test_make_CHS_currents_file.py
index 70ee5633..a731d0e0 100644
--- a/tests/workers/test_make_CHS_currents_file.py
+++ b/tests/workers/test_make_CHS_currents_file.py
@@ -76,7 +76,7 @@ def test_instantiate_worker(self, mock_worker):
         worker = make_CHS_currents_file.main()
         assert worker.name == "make_CHS_currents_file"
         assert worker.description.startswith(
-            "SalishSeaCast worker that average, unstaggers and rotates the near surface velocities,"
+            "SalishSeaCast worker that averages, unstaggers and rotates the near surface velocities,"
         )
 
     def test_add_run_type_arg(self, mock_worker):