test prior fitting

Author: bobmyhill

tests/test_fitting.py

@@ -126,6 +126,35 @@ def test_fit_PVT_data(self):
 
         self.assertArraysAlmostEqual(fitted_eos.pcov[0], zeros)
 
+    def test_fit_PVT_data_w_priors(self):
+        fo = burnman.minerals.HP_2011_ds62.fo()
+
+        pressures = np.linspace(1.0e9, 2.0e9, 8)
+        temperatures = np.ones_like(pressures) * fo.params["T_0"]
+
+        PTV = np.empty((len(pressures), 3))
+
+        for i in range(len(pressures)):
+            fo.set_state(pressures[i], temperatures[i])
+            PTV[i] = [pressures[i], temperatures[i], fo.V]
+
+        params = ["V_0", "K_0", "Kprime_0"]
+        priors = [fo.params["V_0"], fo.params["K_0"], fo.params["Kprime_0"]]
+        invcov = np.linalg.inv(
+            np.diag(np.power(np.array([fo.params["V_0"] / 10.0, 1.0e9, 1.0]), 2.0))
+        )
+        fitted_eos = burnman.eos_fitting.fit_PTV_data(
+            fo,
+            params,
+            PTV,
+            param_priors=priors,
+            param_prior_inv_cov_matrix=invcov,
+            verbose=False,
+        )
+        zeros = np.zeros_like(fitted_eos.pcov[0])
+
+        self.assertArraysAlmostEqual(fitted_eos.pcov[0], zeros)
+
     def test_fit_bounded_PVT_data(self):
         fo = burnman.minerals.HP_2011_ds62.fo()
 
@@ -240,7 +269,14 @@ def test_bounded_solution_fitting(self):
         flags = np.array(flags)
         delta_params = np.array([1.0e-8, 1.0e-8, 1.0e-8])
         bounds = np.array([[0, np.inf], [0, np.inf], [-np.inf, np.inf]])
-
+        priors = np.array(
+            [
+                solution.endmembers[0][0].params["V_0"],
+                solution.endmembers[1][0].params["V_0"],
+                0.0,
+            ]
+        )
+        invcov = np.linalg.inv(np.diag(np.array([1.0e-7, 1.0e-7, 1.0e-7])))
         fitted_eos = fit_XPTp_data(
             solution=solution,
             flags=flags,
@@ -250,6 +286,8 @@ def test_bounded_solution_fitting(self):
             delta_params=delta_params,
             bounds=bounds,
             param_tolerance=1.0e-5,
+            param_priors=priors,
+            param_prior_inv_cov_matrix=invcov,
             verbose=False,
         )