From d19e7325a9aa0d508544a71ea075b61a2bf06832 Mon Sep 17 00:00:00 2001 From: Tim Mensinger Date: Wed, 27 Mar 2024 11:56:32 +0100 Subject: [PATCH] Update tests to include optimize result --- tests/estimation/test_estimate_msm.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/estimation/test_estimate_msm.py b/tests/estimation/test_estimate_msm.py index 6113f265b..6461ef3a5 100644 --- a/tests/estimation/test_estimate_msm.py +++ b/tests/estimation/test_estimate_msm.py @@ -5,6 +5,7 @@ import numpy as np import pandas as pd import pytest +from estimagic.optimization.optimize_result import OptimizeResult from estimagic.estimation.estimate_msm import estimate_msm from estimagic.shared.check_option_dicts import ( check_numdiff_options, @@ -64,6 +65,9 @@ def test_estimate_msm(simulate_moments, moments_cov, optimize_options): # check that minimization works aaae(calculated.params, expected_params) + # assert that optimization result exists and is of correct type + assert isinstance(calculated.optimize_result, OptimizeResult) + # check that cov works calculated_cov = calculated.cov() if isinstance(calculated_cov, pd.DataFrame):