def test_extract_var_data_not_var():
m = ConcreteModel()
with pytest.raises(
ValueError,
match="Unknown variable type of <class "
"'pyomo.core.base.PyomoModel.ConcreteModel'> for unknown"):
_extract_var_data(m)
def test_extract_var_data_indexed_not_ordered():
m = ConcreteModel()
m.s = Set(initialize=[1, 2, 3, 4], ordered=False)
m.v = Var(m.s)
with pytest.raises(ValueError,
match='Expected IndexedVar: v to be indexed over '
'an ordered set.'):
_extract_var_data(m.v)
def test_extract_var_data_multiple():
m = ConcreteModel()
m.s = Set(initialize=[1, 2, 3, 4])
m.vs = Var()
m.vi = Var(m.s)
assert _extract_var_data(
[m.vs, m.vi,
m.vs]) == [m.vs, m.vi[1], m.vi[2], m.vi[3], m.vi[4], m.vs]
def test_extract_var_data_indexed():
m = ConcreteModel()
m.s = Set(initialize=[1, 2, 3, 4])
m.v = Var(m.s)
assert _extract_var_data(m.v) == [m.v[1], m.v[2], m.v[3], m.v[4]]
def test_extract_var_data_scalar():
m = ConcreteModel()
m.v = Var()
assert _extract_var_data(m.v) == [m.v]
def test_extract_var_data_None():
assert _extract_var_data(None) is None