feat: 初始化Easy Patch插件及依赖文件
- 添加Blender插件核心文件:__init__.py、ui.py、property.py、preference.py - 添加插件工具模块:g.py、loop.py、generate_loop.py、const.py、op.py - 添加翻译工具:utils/trans.py - 添加PuLP线性规划库及其依赖文件,包括CBC求解器二进制文件 - 添加.gitignore和VSCode配置文件
This commit is contained in:
122
utils/pulp/tests/test_gurobipy_env.py
Normal file
122
utils/pulp/tests/test_gurobipy_env.py
Normal file
@@ -0,0 +1,122 @@
|
||||
import unittest
|
||||
|
||||
from pulp import GUROBI, LpProblem, LpVariable, const
|
||||
|
||||
try:
|
||||
import gurobipy as gp
|
||||
from gurobipy import GRB
|
||||
except ImportError:
|
||||
gp = None
|
||||
|
||||
|
||||
def check_dummy_env():
|
||||
with gp.Env(params={"OutputFlag": 0}):
|
||||
pass
|
||||
|
||||
|
||||
def generate_lp() -> LpProblem:
|
||||
prob = LpProblem("test", const.LpMaximize)
|
||||
x = LpVariable("x", 0, 1)
|
||||
y = LpVariable("y", 0, 1)
|
||||
z = LpVariable("z", 0, 1)
|
||||
prob += x + y + z, "obj"
|
||||
prob += x + y + z <= 1, "c1"
|
||||
return prob
|
||||
|
||||
|
||||
class GurobiEnvTests(unittest.TestCase):
|
||||
def setUp(self):
|
||||
if gp is None:
|
||||
self.skipTest("Skipping all tests in test_gurobipy_env.py")
|
||||
self.options = {"Method": 0}
|
||||
self.env_options = {"MemLimit": 1, "OutputFlag": 0}
|
||||
|
||||
def test_gp_env(self):
|
||||
# Using gp.Env within a context manager
|
||||
with gp.Env(params=self.env_options) as env:
|
||||
prob = generate_lp()
|
||||
solver = GUROBI(msg=False, env=env, **self.options)
|
||||
prob.solve(solver)
|
||||
solver.close()
|
||||
check_dummy_env()
|
||||
|
||||
@unittest.SkipTest
|
||||
def test_gp_env_no_close(self):
|
||||
# Not closing results in an error for a single use license.
|
||||
with gp.Env(params=self.env_options) as env:
|
||||
prob = generate_lp()
|
||||
solver = GUROBI(msg=False, env=env, **self.options)
|
||||
prob.solve(solver)
|
||||
self.assertRaises(gp.GurobiError, check_dummy_env)
|
||||
|
||||
def test_multiple_gp_env(self):
|
||||
# Using the same env multiple times
|
||||
with gp.Env(params=self.env_options) as env:
|
||||
solver = GUROBI(msg=False, env=env)
|
||||
prob = generate_lp()
|
||||
prob.solve(solver)
|
||||
solver.close()
|
||||
|
||||
solver2 = GUROBI(msg=False, env=env)
|
||||
prob2 = generate_lp()
|
||||
prob2.solve(solver2)
|
||||
solver2.close()
|
||||
|
||||
check_dummy_env()
|
||||
|
||||
@unittest.SkipTest
|
||||
def test_backward_compatibility(self):
|
||||
"""
|
||||
Backward compatibility check as previously the environment was not being
|
||||
freed. On a single-use license this passes (fails to initialise a dummy
|
||||
env).
|
||||
"""
|
||||
solver = GUROBI(msg=False, **self.options)
|
||||
prob = generate_lp()
|
||||
prob.solve(solver)
|
||||
|
||||
self.assertRaises(gp.GurobiError, check_dummy_env)
|
||||
gp.disposeDefaultEnv()
|
||||
solver.close()
|
||||
|
||||
def test_manage_env(self):
|
||||
solver = GUROBI(msg=False, manageEnv=True, **self.options)
|
||||
prob = generate_lp()
|
||||
prob.solve(solver)
|
||||
|
||||
solver.close()
|
||||
check_dummy_env()
|
||||
|
||||
def test_multiple_solves(self):
|
||||
solver = GUROBI(msg=False, manageEnv=True, **self.options)
|
||||
prob = generate_lp()
|
||||
prob.solve(solver)
|
||||
|
||||
solver.close()
|
||||
check_dummy_env()
|
||||
|
||||
solver2 = GUROBI(msg=False, manageEnv=True, **self.options)
|
||||
prob.solve(solver2)
|
||||
|
||||
solver2.close()
|
||||
check_dummy_env()
|
||||
|
||||
@unittest.SkipTest
|
||||
def test_leak(self):
|
||||
"""
|
||||
Check that we cannot initialise environments after a memory leak. On a
|
||||
single-use license this passes (fails to initialise a dummy env with a
|
||||
memory leak).
|
||||
"""
|
||||
solver = GUROBI(msg=False, **self.options)
|
||||
prob = generate_lp()
|
||||
prob.solve(solver)
|
||||
|
||||
tmp = solver.model
|
||||
solver.close()
|
||||
|
||||
solver2 = GUROBI(msg=False, **self.options)
|
||||
|
||||
prob2 = generate_lp()
|
||||
prob2.solve(solver2)
|
||||
self.assertRaises(gp.GurobiError, check_dummy_env)
|
||||
Reference in New Issue
Block a user