Issue #1025 shorten unittests

imod/tests/fixtures/flow_transport_simulation_fixture.py

@@ -210,7 +210,7 @@ def flow_transport_simulation():
     duration = pd.to_timedelta("2000d")
     start = pd.to_datetime("2000-01-01")
     simulation.create_time_discretization(additional_times=[start, start + duration])
-    simulation["time_discretization"]["n_timesteps"] = 100
+    simulation["time_discretization"]["n_timesteps"] = 50
 
     return simulation
     # %%

imod/tests/fixtures/mf6_circle_fixture.py

@@ -74,7 +74,7 @@ def make_circle_model():
         reordering_method=None,
         relaxation_factor=0.97,
     )
-    simtimes = pd.date_range(start="2000-01-01", end="2001-01-01", freq="W")
+    simtimes = pd.date_range(start="2000-01-01", end="2000-01-03")
     simulation.create_time_discretization(additional_times=simtimes)
     return simulation
 
@@ -157,7 +157,7 @@ def make_circle_model_flow_with_transport_data(species: list[str]):
         reordering_method=None,
         relaxation_factor=0.97,
     )
-    simtimes = pd.date_range(start="2000-01-01", end="2001-01-01", freq="W")
+    simtimes = pd.date_range(start="2000-01-01", end="2000-01-03")
     simulation.create_time_discretization(simtimes)
     return simulation
 
@@ -336,7 +336,7 @@ def circle_model_transport():
         reordering_method=None,
         relaxation_factor=0.97,
     )
-    simtimes = pd.date_range(start="2000-01-01", end="2001-01-01", freq="W")
+    simtimes = pd.date_range(start="2000-01-01", end="2000-01-03")
     simulation.create_time_discretization(additional_times=simtimes)
     return simulation
 
@@ -413,6 +413,6 @@ def circle_model_transport_multispecies_variable_density():
         reordering_method=None,
         relaxation_factor=0.97,
     )
-    simtimes = pd.date_range(start="2000-01-01", end="2001-01-01", freq="W")
+    simtimes = pd.date_range(start="2000-01-01", end="2000-01-03")
     simulation.create_time_discretization(additional_times=simtimes)
     return simulation

imod/tests/fixtures/mf6_twri_fixture.py

@@ -227,7 +227,7 @@ def transient_twri_model():
         save_flows=True,
     )
     simulation.create_time_discretization(
-        additional_times=pd.date_range("2000-01-01", " 2000-01-31")
+        additional_times=pd.date_range("2000-01-01", " 2000-01-03")
     )
     return simulation
 
@@ -255,7 +255,7 @@ def transient_unconfined_twri_model():
     # Write specific discharges
     gwf_model["npf"]["save_specific_discharge"] = True
     simulation.create_time_discretization(
-        additional_times=pd.date_range("2000-01-01", " 2000-01-31")
+        additional_times=pd.date_range("2000-01-01", " 2000-01-03")
     )
     return simulation
 

imod/tests/test_mf6/test_mf6_out.py

@@ -222,7 +222,7 @@ def test_open_cbc__dis_transient(transient_twri_result):
             "wel",
         ]
         for array in cbc.values():
-            assert array.shape == (30, 3, 15, 15)
+            assert array.shape == (2, 3, 15, 15)
             assert isinstance(array, xr.DataArray)
             assert isinstance(array.data, dask.array.Array)
 
@@ -266,7 +266,7 @@ def test_open_cbc__dis_transient_unconfined(transient_unconfined_twri_result):
             "wel",
         ]
         for array in cbc.values():
-            assert array.shape == (30, 3, 15, 15)
+            assert array.shape == (2, 3, 15, 15)
             assert isinstance(array, xr.DataArray)
             assert isinstance(array.data, dask.array.Array)
 
@@ -289,10 +289,10 @@ def test_open_cbc__disv(circle_result):
         ]
         for key, array in cbc.items():
             if key in ("chd", "flow-lower-face"):
-                assert array.shape == (52, 2, 216)
+                assert array.shape == (2, 2, 216)
                 assert array.dims[-1] == array.ugrid.grid.face_dimension
             else:
-                assert array.shape == (52, 2, 342)
+                assert array.shape == (2, 2, 342)
                 assert array.dims[-1] == array.ugrid.grid.edge_dimension
             assert isinstance(array, xu.UgridDataArray)
             assert isinstance(array.data, dask.array.Array)
@@ -333,10 +333,10 @@ def test_open_cbc__disv_sto(circle_result_sto):
         ]
         for key, array in cbc.items():
             if key in ("chd", "flow-lower-face", "sto-ss"):
-                assert array.shape == (52, 2, 216)
+                assert array.shape == (2, 2, 216)
                 assert array.dims[-1] == array.ugrid.grid.face_dimension
             else:
-                assert array.shape == (52, 2, 342)
+                assert array.shape == (2, 2, 342)
                 assert array.dims[-1] == array.ugrid.grid.edge_dimension
             assert isinstance(array, xu.UgridDataArray)
             assert isinstance(array.data, dask.array.Array)

imod/tests/test_mf6/test_mf6_simulation.py

@@ -71,15 +71,15 @@ def test_simulation_open_head(circle_model, tmp_path):
 
     assert isinstance(head_notime, xu.UgridDataArray)
     assert head_notime.dims == ("time", "layer", "mesh2d_nFaces")
-    assert head_notime.shape == (52, 2, 216)
+    assert head_notime.shape == (2, 2, 216)
 
     # open heads with time conversion.
     head = simulation.open_head(
         simulation_start_time=datetime(2013, 3, 11, 22, 0, 0), time_unit="w"
     )
     assert head.dims == ("time", "layer", "mesh2d_nFaces")
-    assert head.shape == (52, 2, 216)
-    assert str(head.coords["time"].values[()][0]) == "2013-04-29T22:00:00.000000000"
+    assert head.shape == (2, 2, 216)
+    assert str(head.coords["time"].values[()][0]) == "2013-03-18T22:00:00.000000000"
 
 
 @pytest.mark.usefixtures("circle_model")
@@ -100,7 +100,7 @@ def test_simulation_open_head_relative_path(circle_model, tmp_path):
 
     assert isinstance(head, xu.UgridDataArray)
     assert head.dims == ("time", "layer", "mesh2d_nFaces")
-    assert head.shape == (52, 2, 216)
+    assert head.shape == (2, 2, 216)
 
 
 @pytest.mark.usefixtures("circle_model")

imod/tests/test_mf6/test_multimodel/test_mf6_partitioned_simulation_postprocessing.py

@@ -62,38 +62,7 @@ def test_import_heads_structured(
     assert np.allclose(merged_heads.coords["layer"].values, [1, 2, 3])
     assert np.allclose(
         merged_heads.coords["time"].values,
-        [
-            1.0,
-            2.0,
-            3.0,
-            4.0,
-            5.0,
-            6.0,
-            7.0,
-            8.0,
-            9.0,
-            10.0,
-            11.0,
-            12.0,
-            13.0,
-            14.0,
-            15.0,
-            16.0,
-            17.0,
-            18.0,
-            19.0,
-            20.0,
-            21.0,
-            22.0,
-            23.0,
-            24.0,
-            25.0,
-            26.0,
-            27.0,
-            28.0,
-            29.0,
-            30.0,
-        ],
+        [1.0, 2.0],
     )
 
 
@@ -109,9 +78,7 @@ def test_import_heads_unstructured(tmp_path, circle_partitioned):
 
     # Assert
     assert np.allclose(merged_heads.coords["layer"].values, [1, 2])
-    assert np.allclose(
-        merged_heads.coords["time"].values, list(np.arange(7.0, 365.0, 7.0))
-    )
+    assert np.allclose(merged_heads.coords["time"].values, [1.0, 2.0])
     assert np.allclose(merged_heads.coords["mesh2d_nFaces"].values, list(range(216)))
 
 

imod/tests/test_mf6/test_multimodel/test_mf6_partitioning_unstructured.py

@@ -362,14 +362,14 @@ def run_simulation(tmp_path, simulation, species=None):
     simulation.run()
     head = simulation.open_head()
     flow_budget = simulation.open_flow_budget()
-    flow_budget = flow_budget.sel(time=364)
+    flow_budget = flow_budget.sel(time=2)
     concentration = None
     transport_budget = None
     transport_budget = None
     if has_transport:
         concentration = simulation.open_concentration()
         transport_budget = simulation.open_transport_budget(species)
-        transport_budget = transport_budget.sel(time=364)
+        transport_budget = transport_budget.sel(time=2)
     return head, concentration, flow_budget, transport_budget