def test_bad_repn(self):
repn = list(transforms_nd.registered_transforms.keys())[0]
self.assertTrue(repn in transforms_nd.registered_transforms)
transforms_nd.piecewise_nd(_test_tri, _test_values, repn=repn)
repn = '_bad_repn_'
self.assertFalse(repn in transforms_nd.registered_transforms)
with self.assertRaises(ValueError):
transforms_nd.piecewise_nd(_test_tri, _test_values, repn=repn)
def test_init(self):
for key in transforms_nd.registered_transforms:
for bound in ['lb','ub','eq','bad']:
args = (_test_tri, _test_values)
kwds = {'repn': key, 'bound': bound}
if bound == 'bad':
with self.assertRaises(ValueError):
transforms_nd.piecewise_nd(*args, **kwds)
else:
p = transforms_nd.piecewise_nd(*args, **kwds)
self.assertTrue(
isinstance(p, transforms_nd.registered_transforms[key]))
self.assertTrue(
isinstance(p, TransformedPiecewiseLinearFunctionND))
self.assertEqual(p.active, True)
self.assertIs(p.parent, None)
def test_pickle(self):
for key in transforms_nd.registered_transforms:
p = transforms_nd.piecewise_nd(_test_tri, _test_values, repn=key)
self.assertEqual(p.parent, None)
pup = pickle.loads(pickle.dumps(p))
self.assertEqual(pup.parent, None)
b = block()
b.p = p
self.assertIs(p.parent, b)
bup = pickle.loads(pickle.dumps(b))
pup = bup.p
self.assertIs(pup.parent, bup)
def test_type(self):
for key in transforms_nd.registered_transforms:
p = transforms_nd.piecewise_nd(_test_tri, _test_values, repn=key)
# small block storage
self.assertTrue(len(list(p.children())) <= 4)
self.assertTrue(isinstance(p,
TransformedPiecewiseLinearFunctionND))
self.assertTrue(
isinstance(p, transforms_nd.registered_transforms[key]))
self.assertTrue(isinstance(p, ICategorizedObject))
self.assertTrue(isinstance(p, ICategorizedObjectContainer))
self.assertTrue(isinstance(p, IHeterogeneousContainer))
self.assertTrue(isinstance(p, IBlock))
self.assertTrue(isinstance(p, block))
class Test_piecewise_nd_list(_TestActiveListContainerBase, unittest.TestCase):
_container_type = block_list
_ctype_factory = lambda self:\
transforms_nd.piecewise_nd(_test_tri,
_test_values)
