B u `9@sddgZddlZddlZddlmZddlmZddlmZddl Z ddl m Z ddl mZdd lmZiZd d Zd dZdS) test_solver_casesZavailable_solversN)Options) SolverFactory) UnknownSolver) GUROBISHELL) BARONSHELL) MOSEKDirectcKstf|}d|_|jdkr(ts(d|_d}yt|j|jd}Wn YnX|dks`t|trhd|_ nZ|jdkrt sd|_ n@|jdkrt sd|_ n&|j ddot |d  p| dk |_ |dkst|tsx*|jD] }||jkrtd |j|fqW|j r||_|S) N)NNNbaron) r 2)Z solver_ioFgurobi> mosek_directmosek_persistent)Zexception_flag executablez)Solver %s does not support capability %s!)rZ demo_limitsnamerZlicense_is_validrio isinstancer availablerrhasattrr capabilitiesZ _capabilities ValueErrorversion)kwdsobjoptZ_cr?/tmp/pip-unpacked-wheel-n62dbgi3/pyomo/solvers/tests/solvers.py initialize$s8         rcGsttdkrttjtdddddg}tdd|d d d gd td <tdd|d d d gd td<tddddddg}tdd|d d d gd td<tdd|d d d gd td<tdd|d gd td<tdd|d d d gd td<tdd|d d d gd td<tddddg}tdd|d d gd td<tdd|d d gd td<tddddddg}td d|d d d gd td!<td d|d d d gd td"<td d|d#d#d$d gd%td&<td d|d d d gd td'<td(d|d d d gd td)<tddg}d*ttj j j j j krd g}nd d g}td+d||d td,<td+d||d-d.id/td0<td+d|gd td1<tddg}td2d|d d gd td3<tddddg}td2d|d gd td4<tddg} td5d| d gd td6<td5d| d gd td7<tddddddg} td8d| d d d gd9d:id;td<<td8d| d d d gd9d:id;td=<td8d| d gd9d:id;td><td8d| d d d gd9d:id;td?<td@d| d d d gd9d:id;tdA<tdddg} tdBd| d gd tdC<tddddddg} tdDd| gd tdE<tddddddg} tdFd| gd tdG<tddddg}tdHdI|d d gd tdJ<tddddg}tdKd|d gd tdL<ttj xXttD]H}|jdMkrptg|_|jdMkrg|_|jdMkrg|_|jdMkri|_|jdMkri|_|jdMk rt|jtkstt|jttgkstt|jttgkstt|jtkstx"|jD]}t|tks$tq$Wx"|jD]}t|tksHtqHWt|jtksxtx"|jD]}t|tkstqWqTWt|dkrtSt|S)NzF A function for accessing _test_solver_casess as global state rZlinearintegerZquadratic_objectiveZquadratic_constraintZconic_constraintsr pythonZdualrcZslack)rrrimport_suffixes)Zmosekrr)rrZsos1Zsos2cplexlp)r"r#mps)r"r$nl)r"r%)r"rcplex_persistent)r&rgamsgms)r'r()r'rr )r r#)r r$)ZqcpdualZsimplex)rrroptionsr!)r r%)r rgurobi_persistent)r+rZ GLPKSHELL_oldglpk)r,r#Zskip_objective_senseT)rrrr! io_options)r,r$)r,rcbc)r.r#)r.r%pico)r/r#)r/r%xpressZ bargapstopg& .>)rrrr!r*)r0r#)r0r$)r0r%)r0rxpress_persistent)r1ripopt)r2r%scip)r3r%conopt)r4r%r bar)r r5 knitroampl)r6r%N)len_test_solver_casesloggingdisableWARNINGsetrstrpyomoZsolversZpluginsZGLPK __class__NOTSETsix itervaluesrZexport_suffixesr!r*r-rtypeAssertionErrorlisttupledictkeys)argsZ_mosek_capabilitiesZ_cplex_capabilitiesZ_gams_capabilitiesZ_gurobi_capabilitiesZ_glpk_capabilitiesZglpk_import_suffixesZ_cbc_lp_capabilitiesZ_cbc_nl_capabilitiesZ_pico_capabilitiesZ_xpress_capabilitiesZ_ipopt_capabilitiesZ_scip_capabilitiesZ_conopt_capabilitiesZ_baron_capabilitiesZ_knitroampl_capabilitiesZsctagrrrrRs                  )__all__rAr9Zpyomo.common.collectionsrZ pyomo.optrZpyomo.opt.base.solversrZ pyomo.environr>Z$pyomo.solvers.plugins.solvers.GUROBIrZ#pyomo.solvers.plugins.solvers.BARONrZ*pyomo.solvers.plugins.solvers.mosek_directrr8rrrrrr s       .