Fixed test for 1691 multi-chain

pints-team · Dec 18, 2024 · c4c76a6 · c4c76a6
1 parent 30eec62
commit c4c76a6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pints/tests/test_mcmc_controller.py b/pints/tests/test_mcmc_controller.py
@@ -768,6 +768,11 @@ def test_log_pdf_storage_in_memory_multi(self):
         mcmc.set_log_to_screen(False)
         mcmc.set_log_pdf_storage(True)
         chains = mcmc.run()
+        evals = mcmc.log_pdfs()
+        self.assertEqual(len(evals.shape), 3)
+        self.assertEqual(evals.shape[0], n_chains)
+        self.assertEqual(evals.shape[1], n_iterations)
+        self.assertEqual(evals.shape[2], 3)
 
         # Test with a loglikelihood
         mcmc = pints.MCMCController(