def test_vecwrapper(self):
unknowns_dict = OrderedDict()
unknowns_dict["y1"] = {"shape": (3, 2), "size": 6, "val": np.ones((3, 2))}
unknowns_dict["y2"] = {"shape": 1, "size": 1, "val": 2.0}
unknowns_dict["y3"] = {"size": 0, "val": "foo", "pass_by_obj": True}
unknowns_dict["y4"] = {"shape": (2, 1), "size": 2, "val": np.zeros((2, 1))}
unknowns_dict["s1"] = {"shape": 1, "size": 1, "val": -1.0, "state": True}
sd = _SysData("")
for u, meta in unknowns_dict.items():
meta["pathname"] = u
meta["top_promoted_name"] = u
sd.to_prom_name[u] = u
u = SrcVecWrapper(sd)
u.setup(unknowns_dict, store_byobjs=True)
self.assertEqual(u.vec.size, 10)
self.assertEqual(len(u), 5)
self.assertEqual(list(u.keys()), ["y1", "y2", "y3", "y4", "s1"])
self.assertTrue(np.all(u["y1"] == np.ones((3, 2))))
self.assertEqual(u["y2"], 2.0)
self.assertEqual(u["y3"], "foo")
self.assertTrue(np.all(u["y4"] == np.zeros((2, 1))))
self.assertEqual(u["s1"], -1.0)
self.assertEqual(u.get_states(), ["s1"])
self.assertEqual([t[0] for t in u.veciter()], ["y1", "y2", "y4", "s1"])
u["y1"] = np.ones((3, 2)) * 3.0
u["y2"] = 2.5
u["y3"] = "bar"
u["y4"] = np.ones((2, 1)) * 7.0
u["s1"] = 5.0
self.assertTrue(np.all(u["y1"] == np.ones((3, 2)) * 3.0))
self.assertTrue(np.all(u["y4"] == np.ones((2, 1)) * 7.0))
self.assertEqual(u["y2"], 2.5)
self.assertEqual(u["y3"], "bar")
self.assertEqual(u["s1"], 5.0)
# set with a different shaped array
try:
u["y1"] = np.ones((3, 3))
except Exception as err:
self.assertEqual(str(err), "could not broadcast input array from shape (9) into shape (6)")
else:
self.fail("Exception expected")
params = OrderedDict()
params["y1"] = {"shape": (3, 2), "size": 6, "val": np.ones((3, 2))}
params["y2"] = {"shape": 1, "size": 1, "val": 2.0}
params["y3"] = {"size": 0, "val": "foo"}
params["y4"] = {"shape": (2, 1), "size": 6, "val": np.zeros((2, 1))}
for p, meta in params.items():
meta["pathname"] = p
meta["top_promoted_name"] = p
sd.to_prom_name[u] = u
connections = {}
for p in params:
connections[p] = (p, None)
s = _SysData("")
s._unknowns_dict = u._dat
p = TgtVecWrapper(s)
p.setup(None, params, u, params.keys(), connections, store_byobjs=True)
self.assertEqual(p.vec.size, 9)
self.assertEqual(len(p), 4)
self.assertEqual(list(p.keys()), ["y1", "y2", "y3", "y4"])
self.assertTrue(np.all(p["y1"] == np.zeros((3, 2))))
self.assertEqual(p["y2"], 0.0)
self.assertEqual(p["y3"], "bar")
self.assertTrue(np.all(p["y4"] == np.zeros((2, 1))))
p["y1"] = np.ones((3, 2)) * 9.0
self.assertTrue(np.all(p["y1"] == np.ones((3, 2)) * 9.0))
def test_vecwrapper(self):
unknowns_dict = OrderedDict()
unknowns_dict['y1'] = { 'shape': (3,2), 'size': 6, 'val': np.ones((3, 2)) }
unknowns_dict['y2'] = { 'shape': 1, 'size': 1, 'val': 2.0 }
unknowns_dict['y3'] = { 'size': 0, 'val': "foo", 'pass_by_obj': True }
unknowns_dict['y4'] = { 'shape': (2,1), 'size': 2, 'val': np.zeros((2, 1)), }
unknowns_dict['s1'] = { 'shape': 1, 'size': 1, 'val': -1.0, 'state': True, }
for u, meta in unknowns_dict.items():
meta['pathname'] = u
meta['promoted_name'] = u
meta['top_promoted_name'] = u
u = SrcVecWrapper(_SysData(''))
u.setup(unknowns_dict, store_byobjs=True)
self.assertEqual(u.vec.size, 10)
self.assertEqual(len(u), 5)
self.assertEqual(list(u.keys()), ['y1','y2','y3', 'y4', 's1'])
self.assertTrue(np.all(u['y1']==np.ones((3,2))))
self.assertEqual(u['y2'], 2.0)
self.assertEqual(u['y3'], 'foo')
self.assertTrue(np.all(u['y4']==np.zeros((2,1))))
self.assertEqual(u['s1'], -1.0)
self.assertEqual(u.get_states(), ['s1'])
self.assertEqual([t[0] for t in iteritems(u.flat)], ['y1','y2','y4','s1'])
self.assertEqual([t[0] for t in u.get_byobjs()], ['y3'])
u['y1'] = np.ones((3,2))*3.
u['y2'] = 2.5
u['y3'] = 'bar'
u['y4'] = np.ones((2,1))*7.
u['s1'] = 5.
self.assertTrue(np.all(u['y1']==np.ones((3,2))*3.))
self.assertTrue(np.all(u['y4']==np.ones((2,1))*7.))
self.assertEqual(u['y2'], 2.5)
self.assertEqual(u['y3'], 'bar')
self.assertEqual(u['s1'], 5.)
# set with a different shaped array
try:
u['y1'] = np.ones((3,3))
except Exception as err:
self.assertEqual(str(err),
"could not broadcast input array from shape (9) into shape (6)")
else:
self.fail("Exception expected")
params = OrderedDict()
params['y1'] = { 'shape': (3,2), 'size': 6, 'val': np.ones((3, 2)) }
params['y2'] = { 'shape': 1, 'size': 1, 'val': 2.0 }
params['y3'] = { 'size': 0, 'val': "foo" }
params['y4'] = { 'shape': (2,1), 'size': 6, 'val': np.zeros((2, 1)) }
for p, meta in params.items():
meta['pathname'] = p
meta['promoted_name'] = p
meta['top_promoted_name'] = p
connections = {}
for p in params:
connections[p] = (p, None)
s = _SysData('')
s._unknowns_dict = u._vardict
p = TgtVecWrapper(s)
p.setup(None, params, u, params.keys(),
connections, store_byobjs=True)
self.assertEqual(p.vec.size, 9)
self.assertEqual(len(p), 4)
self.assertEqual(list(p.keys()), ['y1','y2','y3', 'y4'])
self.assertTrue(np.all(p['y1']==np.zeros((3,2))))
self.assertEqual(p['y2'], 0.)
self.assertEqual(p['y3'], 'bar')
self.assertTrue(np.all(p['y4']==np.zeros((2,1))))
p['y1'] = np.ones((3,2))*9.
self.assertTrue(np.all(p['y1']==np.ones((3,2))*9.))