def test_init(self):
var = Var(type='test_type',
value='test_value',
source_tag='source_tag')
self.assertEqual(var.type, 'test_type')
self.assertEqual(var.value, 'test_value')
self.assertEqual(var.source_tag, 'source_tag')
def test_to_dict(self):
params = Params()
params.params['${constant_1}'] = Var(type=Var.PLAIN, value='value_1')
params.params['${constant_2}'] = {'type': 'plain', 'value': 'value_2'}
d = params.to_dict()
self.assertEqual(d, {'${constant_1}': {'type': 'plain', 'value': 'value_1'},
'${constant_2}': {'type': 'plain', 'value': 'value_2'}})
def test_to_dict(self):
splice_var = Var(type=Var.SPLICE, value='val', source_tag='source_tag')
plain_var = Var(type=Var.PLAIN, value='val', source_tag='source_tag')
lazy_var = Var(type=Var.LAZY, value='val', source_tag='source_tag')
self.assertEqual(splice_var.to_dict(), {
'type': Var.SPLICE,
'value': 'val'
})
self.assertEqual(plain_var.to_dict(), {
'type': Var.PLAIN,
'value': 'val'
})
self.assertEqual(lazy_var.to_dict(), {
'type': Var.LAZY,
'value': 'val',
'source_tag': 'source_tag'
})
def test_to_dict(self):
data = Data()
data.inputs['${constant_1}'] = Var(type=Var.PLAIN, value='value_1')
data.inputs['${constant_2}'] = {'type': 'plain', 'value': 'value_2'}
data.outputs.append('${constant_1}')
d = data.to_dict()
self.assertEqual(
d, {
'inputs': {
'${constant_1}': {
'type': 'plain',
'value': 'value_1'
},
'${constant_2}': {
'type': 'plain',
'value': 'value_2'
}
},
'outputs': ['${constant_1}']
})