bi^dZddlZddlZddlmZddlmZejZ GddeZ y)a Tests for DGP (Disciplined Geometric Programming) with DPP (Disciplined Parametrized Programming). This verifies the fix for issue #3004: DGP problems can now have get_problem_data(gp=True) called without all parameters having values. This is useful when the problem is DPP. N)ParameterError)BaseTestc0eZdZdZddZddZddZddZy) TestDgpDppzPTests for DGP + DPP integration focusing on get_problem_data core functionality.Ncjtjd}tjd}tjtj|||k\g}|j t d\}}}|Jd|_|jt dd|j|jdy)zget_problem_data(gp=True) works with uninitialized parameters. This was the original failure case in issue #3004. Previously, DGP canonicalization eagerly evaluated np.log(param.value), failing when param.value was None. TposgpN@r enforce_dpp) cp ParameterVariableProblemMinimizeget_problem_dataSOLVERvaluesolveassertAlmostEqual)selfalphaxproblemdata_s S/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/tests/test_dgp_dpp.py*test_get_problem_data_without_param_valuesz5TestDgpDpp.test_get_problem_data_without_param_valuess & KKD !**R[[^a5j\:--f-> a  f4 8 qww,c$tjd}tjd}tjtj|||k\g}d|_|j tdd|j|j ddd|_|j tdd|j|j ddd|_|j tdd|j|j ddy ) zDGP DPP fast path correctly updates parameters across solves. Tests that log-transformed parameter values are properly updated in the cached param_prog on each solve. Tr?r placesr g?N) rrrrrrrrrrrrrs r#test_fast_path_with_changing_paramsz.TestDgpDpp.test_fast_path_with_changing_params's  & KKD !**R[[^a5j\:  f4 8 qwwA6  f4 8 qwwA6  f4 8 qwwA6r!cttjd}tjd}tjtj|||k\g}|j t dtjtd5|jt dddddy#1swYyxYw)zKAttempting to solve without setting parameter values raises ParameterError.Trr z must have.*value.*before solving)matchr N) rrrrrrrpytestraisesrrr's r+test_solve_without_param_value_raises_errorz6TestDgpDpp.test_solve_without_param_value_raises_error@s & KKD !**R[[^a5j\:   D 1]]>1S T = MM&TtM < = = =s B..B7ctjdd}tjd}tjtj|||k\g}|j dsJ|j tdd|j|jddd |_ |j tdd|j|jd dy ) a$DGP problems work correctly with ignore_dpp=True (non-DPP mode). This tests backward compatibility: even though the DGP canonicalization now supports DPP, forcing non-DPP treatment with ignore_dpp=True should still work correctly when parameters have values. Tr#)r rrdgp)r ignore_dppr$r%g@N) rrrrris_dpprrrrr's r&test_non_dpp_mode_with_ignore_dpp_flagz1TestDgpDpp.test_non_dpp_mode_with_ignore_dpp_flagMs S1 KKD !**R[[^a5j\:~~e$$$ f$ 7 qwwA6  f$ 7 qwwA6r!)returnN)__name__ __module__ __qualname____doc__r r(r-r2r!rrrsZ-(72 =7r!r) r7r+cvxpyr cvxpy.errorrcvxpy.tests.base_testrCLARABELrrr8r!rr=s/ &* P7P7r!