Add unitest

Author: domfournier

tests/driver_test.py

@@ -57,3 +57,40 @@ def test_smallness_terms(tmp_path: Path):
         params.alpha_s = None
         driver = GravityInversionDriver(params)
         assert driver.regularization.objfcts[0].alpha_s == 0.0
+
+
+def test_target_chi(tmp_path: Path, caplog):
+    n_grid_points = 2
+    refinement = (2,)
+
+    geoh5, _, model, survey, topography = setup_inversion_workspace(
+        tmp_path,
+        background=0.0,
+        anomaly=0.75,
+        n_electrodes=n_grid_points,
+        n_lines=n_grid_points,
+        refinement=refinement,
+        flatten=False,
+    )
+
+    with geoh5.open():
+        gz = survey.add_data({"gz": {"values": np.ones(survey.n_vertices)}})
+        mesh = model.parent
+        active_cells = ActiveCellsOptions(topography_object=topography)
+        params = GravityInversionOptions(
+            geoh5=geoh5,
+            mesh=mesh,
+            active_cells=active_cells,
+            data_object=gz.parent,
+            gz_channel=gz,
+            gz_uncertainty=2e-3,
+            starting_model=1e-4,
+            starting_chi_factor=1.0,
+            chi_factor=2.0,
+        )
+        driver = GravityInversionDriver(params)
+
+        with caplog.at_level("WARNING"):
+            assert driver.directives.update_irls_directive.chifact_start == 2.0
+
+        assert "Starting chi factor is greater" in caplog.text