Basic test

.gitignore

@@ -172,3 +172,5 @@ cython_debug/
 
 # Auto generated by setuptools scm
 virtual_ship/_version_setup.py
+
+.vscode/

pyproject.toml

@@ -53,6 +53,7 @@ dev = [
     "isort == 5.13.2",
     "pydocstyle == 6.3.0",
     "sort-all == 1.2.0",
+    "pytest == 8.2.0",
 ]
 
 [tool.isort]

tests/conftest.py

@@ -0,0 +1,7 @@
+import pytest
+
+
+# Set the working directory for each test to the directory of that test.
+@pytest.fixture(autouse=True)
+def change_test_dir(request, monkeypatch):
+    monkeypatch.chdir(request.fspath.dirname)

tests/instruments/test_argo.py

@@ -0,0 +1,40 @@
+"""Test the simulation of argos."""
+
+from virtual_ship.instruments.argo import simulate_argos, Argo
+from virtual_ship.instruments.location import Location
+from parcels import FieldSet
+import numpy as np
+
+
+def test_simulate_argos() -> None:
+    DRIFT_DEPTH = -1000
+    MAX_DEPTH = -2000
+    VERTICLE_SPEED = -0.10
+    CYCLE_DAYS = 10
+    DRIFT_DAYS = 9
+
+    fieldset = FieldSet.from_data(
+        {"U": 0, "V": 0, "T": 0, "S": 0},
+        {
+            "lon": 0,
+            "lat": 0,
+            "time": [np.datetime64("1950-01-01") + np.timedelta64(632160, "h")],
+        },
+    )
+
+    min_depth = -fieldset.U.depth[0]
+
+    argos = [
+        Argo(
+            location=Location(latitude=0, longitude=0),
+            deployment_time=0,
+            min_depth=min_depth,
+            max_depth=MAX_DEPTH,
+            drift_depth=DRIFT_DEPTH,
+            vertical_speed=VERTICLE_SPEED,
+            cycle_days=CYCLE_DAYS,
+            drift_days=DRIFT_DAYS,
+        )
+    ]
+
+    simulate_argos(argos=argos, fieldset=fieldset, out_file_name="test")

tests/sailship_config.json

@@ -0,0 +1,40 @@
+{
+    "region_of_interest": {
+        "North": 64,
+        "East": -23,
+        "South": 59,
+        "West": -43
+      },
+    "requested_ship_time": {
+        "start": "2022-01-01T00:00:00",
+        "end": "2022-01-2T00:00:00"
+      },
+    "route_coordinates": [
+      [-23.071289, 63.743631], [-23.081289, 63.743631], [-23.091289, 63.743631]
+    ],
+    "underway_data": true,
+    "ADCP_data": false,
+    "ADCP_settings": {
+        "max_depth": -1000,
+        "bin_size_m": 24
+      },
+    "CTD_locations": [
+      [-23.071289, 63.743631]
+    ],
+    "CTD_settings": {
+        "max_depth": "max"
+      },
+    "drifter_deploylocations": [
+
+    ],
+    "argo_deploylocations": [
+      [-23.081289, 63.743631]
+    ],
+    "argo_characteristics": {
+      "driftdepth": -1000,
+      "maxdepth": -2000,
+      "vertical_speed": -0.10, 
+      "cycle_days" : 10, 
+      "drift_days": 9
+    }
+  }

tests/test_sailship.py

@@ -0,0 +1,42 @@
+"""Performs a complete cruise with virtual ship."""
+
+from virtual_ship.virtual_ship_configuration import VirtualShipConfiguration
+from virtual_ship.sailship import sailship
+from parcels import FieldSet
+import numpy as np
+
+
+def test_sailship() -> None:
+    ctd_fieldset = FieldSet.from_data(
+        {"U": 0, "V": 0, "S": 0, "T": 0, "bathymetry": 0},
+        {"lon": 0, "lat": 0},
+    )
+    ctd_fieldset.add_constant("maxtime", ctd_fieldset.U.grid.time_full[-1])
+    ctd_fieldset.add_constant("mindepth", -ctd_fieldset.U.depth[0])
+    ctd_fieldset.add_constant("max_depth", -ctd_fieldset.U.depth[-1])
+
+    drifter_fieldset = FieldSet.from_data(
+        {
+            "U": 0,
+            "V": 0,
+        },
+        {"lon": 0, "lat": 0},
+    )
+
+    argo_fieldset = FieldSet.from_data(
+        {"U": 0, "V": 0, "T": 0, "S": 0},
+        {
+            "lon": 0,
+            "lat": 0,
+            "time": [np.datetime64("1950-01-01") + np.timedelta64(632160, "h")],
+        },
+    )
+
+    config = VirtualShipConfiguration(
+        "sailship_config.json",
+        ctd_fieldset=ctd_fieldset,
+        drifter_fieldset=drifter_fieldset,
+        argo_fieldset=argo_fieldset,
+    )
+
+    sailship(config)

virtual_ship/drifter_deployments.py

@@ -7,8 +7,10 @@
 import numpy as np
 from parcels import AdvectionRK4, FieldSet, JITParticle, ParticleSet, Variable
 
+from .virtual_ship_configuration import VirtualShipConfiguration
 
-def drifter_deployments(config, drifter_time):
+
+def drifter_deployments(config: VirtualShipConfiguration, drifter_time):
     """
     Deploy drifters.
 
@@ -17,7 +19,7 @@ def drifter_deployments(config, drifter_time):
     """
     if len(config.drifter_deploylocations) > 0:
 
-        fieldset = create_drifter_fieldset(config)
+        fieldset = config.drifter_fieldset
 
         # Create particle to sample water underway
         class DrifterParticle(JITParticle):
@@ -74,18 +76,18 @@ def CheckError(particle, fieldset, time):
         )
 
 
-def create_drifter_fieldset(config):
+def create_drifter_fieldset(config, data_dir: str):
     """
     Create a fieldset from netcdf files for drifters, returns fieldset with negative depth values.
 
     :param config: The cruise configuration.
+    :param data_dir: TODO
     :returns: The fieldset.
     """
-    datadirname = os.path.dirname(__file__)
     filenames = {
-        "U": os.path.join(datadirname, "drifterdata_UV.nc"),
-        "V": os.path.join(datadirname, "drifterdata_UV.nc"),
-        "T": os.path.join(datadirname, "drifterdata_T.nc"),
+        "U": os.path.join(data_dir, "drifterdata_UV.nc"),
+        "V": os.path.join(data_dir, "drifterdata_UV.nc"),
+        "T": os.path.join(data_dir, "drifterdata_T.nc"),
     }
     variables = {"U": "uo", "V": "vo", "T": "thetao"}
     dimensions = {