def test_parameter_single(self):
# init
p0 = Parameter('param', 0, int, 'integer')
# print(p0.info)
self.assertEqual(p0.info['param'], 'integer')
p0 = Parameter('param', 0.0, float, 'float')
p0 = Parameter('param', '0', str, 'string')
p0 = Parameter('param', 0, [0,1,2,3], 'list')
p0 = Parameter('param', 0)
self.assertEqual(p0.valid_values['param'], int)
p0 = Parameter('param', 0.0)
self.assertEqual(p0.valid_values['param'], float)
p0 = Parameter('param', '0')
self.assertEqual(p0.valid_values['param'], str)
p0 = Parameter('param', [0,1,2,3])
self.assertEqual(p0.valid_values['param'], list)
p0 = Parameter('param', 0, int, 'integer')
self.assertEqual(p0,{'param':0})
self.assertEqual(p0['param'],0)
p0 = Parameter({'param': 1})
self.assertEqual(p0,{'param':1})
#update
p0.update({'param':2})
self.assertEqual(p0,{'param':2})
with self.assertRaises(KeyError):
p0.update({'paramX':2})
with self.assertRaises(AssertionError):
Parameter('param1', 0, [ 1, 2, 3])
with self.assertRaises(AssertionError):
Parameter('pa',2.2, int, 'type is int but value is float!')
p0 = Parameter('param', [0,1,2,3])
p0.update({'param':[0,5]})
with self.assertRaises(AssertionError):
p0.update({'param':0})
def test_parameter_multi(self):
# init
p1 = Parameter('param1', 1)
p2 = Parameter('param2', 2)
p0 = Parameter('param0', [p1, p2])
self.assertEqual(p0 , {'param0': {'param1':1, 'param2':2}})
#update
p0['param0'] = {'param1':3}
self.assertEqual(p0 , {'param0': {'param1':3, 'param2':2}})
self.assertEqual(p0['param0'] , {'param1':3, 'param2':2})
self.assertEqual(p0['param0']['param1'] , 3)
with self.assertRaises(KeyError):
p0.update({'param1':4})
p0['param0'].update(Parameter('param2', 7))
self.assertEqual(p0['param0'], {'param1':3, 'param2':7})
p0['param0'].update({'param2': 8})
self.assertEqual(p0['param0'] , {'param1':3, 'param2':8})
p0['param0'] = {'param1':5, 'param2':6}
self.assertEqual(p0['param0'] , {'param1':5, 'param2':6})
self.assertEqual(p0['param0']['param1'] ,5)
self.assertEqual(p0['param0']['param2'] ,6)
# p0['param0'].update(Parameter('param2', 's'))
#
# p0['param0'].update(Parameter('param3', 's'))
#
# p0['param0'].update({'param2', 's'})
with self.assertRaises(KeyError):
print((p0['param3']))
p1 = Parameter('param1', 1)
p2 = Parameter('param2', 2)
p0 = Parameter('param0', [p1, p2])
with self.assertRaises(AssertionError):
p0['param0'] = [0,1] # asign list of different type
p1 = Parameter('param1', 1)
p2 = Parameter('param2', 2)
p3 = Parameter('param2', 3)
p0 = Parameter('param0', [p1, p2])
p0['param0'] = [p1,p3]
with self.assertRaises(AssertionError):
p1 = Parameter('param1', 1)
p2 = Parameter('param2', 2)
p3 = Parameter('param3', 3)
p0 = Parameter('param0', [p1, p2])
p0['param0'] = [p1,p3]
with self.assertRaises(AssertionError):
p1 = Parameter('param1', 1)
p2 = Parameter('param2', 2)
p3 = Parameter('param2', 's')
p0 = Parameter('param0', [p1, p2])
p0['param0'] = [p1,p3]
def test_parameter_multi_v2(self):
'''
test for next generation parameter object
Args:
self:
Returns:
'''
p0 = Parameter({'p1' : 1, 'p2' : 2})
self.assertEqual(p0, {'p2': 2, 'p1': 1})
with self.assertRaises(KeyError):
p0['p3']
with self.assertRaises(KeyError):
p0.update({'p3': 2})
with self.assertRaises(AssertionError):
p0.update({'p1': 2.0})
p0 = Parameter('p0', 0)
p1 = Parameter('p1', 1)
p2 = Parameter([p0,p1])
self.assertEqual(p2, {'p0': 0, 'p1': 1})
p2['p0'] = 44
self.assertEqual(p2, {'p0': 44, 'p1': 1})
p2.update({'p0': 45 })
self.assertEqual(p2, {'p0': 45, 'p1': 1})
p0['p0'] = 555
p2.update(p0)
self.assertEqual(p2, {'p0': 555, 'p1': 1})
# check for nested parameters
p0 = Parameter('p0', 0, int, 'test parameter (int)')
p1 = Parameter('p1', 'string', str, 'test parameter (str)')
p2 = Parameter('p2', 0.0, float, 'test parameter (float)')
p12 = Parameter('p12', [p1,p2])
pall = Parameter([p0, p12])
self.assertEqual(pall, {'p0': 0, 'p12': {'p2': 0.0, 'p1': 'string'}})
parameters = Parameter([
Parameter('test1', 0, int, 'test parameter (int)'),
Parameter('test2' ,
[Parameter('test2_1', 'string', str, 'test parameter (str)'),
Parameter('test2_2', 0.0, float, 'test parameter (float)')
])
])
self.assertIsInstance(parameters['test2'], Parameter)
with self.assertRaises(AssertionError):
parameters['test1'] = 0.2
with self.assertRaises(AssertionError):
parameters['test2'] = 0.2
with self.assertRaises(AssertionError):
parameters['test2']['test2_1'] = 0.2
with self.assertRaises(AssertionError):
parameters['test2']['test2_2'] = 's'
