Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add is_reference API #1740

Merged
merged 12 commits into from
Jan 19, 2021
Merged

Add is_reference API #1740

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
aa770b3
referent attribute
Robbybp Dec 9, 2020
064c51f
base class is_reference method
Robbybp Dec 9, 2020
492fb91
implement is_reference and fix for slow iteration over ref-to-ref
Robbybp Dec 9, 2020
c26d16d
change method to owns_data
Robbybp Dec 28, 2020
a3f23c9
Merge branch 'master' of https://github.com/Pyomo/pyomo into referenc…
Robbybp Dec 28, 2020
af7330c
negate logic for owns_data
Robbybp Dec 28, 2020
4a73994
assign referent before processing argument type
Robbybp Dec 28, 2020
84a3943
test to cover base class owns_data
Robbybp Dec 28, 2020
42eba1a
tests for owns_data and referent attribute
Robbybp Dec 28, 2020
d886b72
Merge branch 'master' of https://github.com/Pyomo/pyomo into referenc…
Robbybp Jan 14, 2021
98eb6fd
change name back to is_reference
Robbybp Jan 14, 2021
778aee5
expand docstring in is_reference
Robbybp Jan 16, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
tests for owns_data and referent attribute
  • Loading branch information
@Robbybp
Robbybp committed Dec 28, 2020
commit 42eba1ad7fe78e8495a29cbfb0057837be76f234
52 changes: 52 additions & 0 deletions pyomo/core/tests/unit/test_reference.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
from pyomo.core.base.indexed_component import (
UnindexedComponent_set, IndexedComponent
)
from pyomo.core.base.indexed_component_slice import IndexedComponent_slice
from pyomo.core.base.reference import (
_ReferenceDict, _ReferenceSet, Reference
)
Expand Down Expand Up @@ -803,6 +804,57 @@ def test_reference_to_list(self):
KeyError, "Index '1' is not valid for indexed component 'r'"):
m.r[1] = m.x

def test_owns_data(self):
m = ConcreteModel()
m.v0 = Var()
m.v1 = Var([1,2,3])

m.ref0 = Reference(m.v0)
m.ref1 = Reference(m.v1)

self.assertTrue(m.v0.owns_data())
self.assertTrue(m.v1.owns_data())

self.assertFalse(m.ref0.owns_data())
self.assertFalse(m.ref1.owns_data())

unique_vars = list(
v for v in m.component_objects(Var) if v.owns_data())
self.assertEqual(len(unique_vars), 2)

def test_referent(self):
m = ConcreteModel()
m.v0 = Var()
m.v2 = Var([1, 2, 3],['a', 'b'])

varlist = [m.v2[1, 'a'], m.v2[1, 'b']]

vardict = {
0: m.v0,
1: m.v2[1, 'a'],
2: m.v2[2, 'a'],
3: m.v2[3, 'a'],
}

scalar_ref = Reference(m.v0)
self.assertIs(scalar_ref.referent, m.v0)

sliced_ref = Reference(m.v2[:,'a'])
referent = sliced_ref.referent
self.assertIs(type(referent), IndexedComponent_slice)
self.assertEqual(len(referent._call_stack), 1)
call, info = referent._call_stack[0]
self.assertEqual(call, IndexedComponent_slice.slice_info)
self.assertIs(info[0], m.v2)
self.assertEqual(info[1], {1: 'a'}) # Fixed
self.assertEqual(info[2], {0: slice(None)}) # Sliced
self.assertIs(info[3], None) # Ellipsis

list_ref = Reference(varlist)
self.assertIs(list_ref.referent, varlist)

dict_ref = Reference(vardict)
self.assertIs(dict_ref.referent, vardict)

if __name__ == "__main__":
unittest.main()