def test_build_spec(self):
spec_dict = {
'name': 'attribute1',
'doc': 'my first attribute',
'dtype': 'text',
'shape': [None],
'dims': ['dim1'],
'value': ['a', 'b']
}
ret = AttributeSpec.build_spec(spec_dict)
self.assertTrue(isinstance(ret, AttributeSpec))
self.assertDictEqual(ret, spec_dict)
def test_build_spec_reftype(self):
spec_dict = {
'name': 'attribute1',
'doc': 'my first attribute',
'dtype': {
'target_type': 'AnotherType',
'reftype': 'object'
}
}
expected = spec_dict.copy()
expected['dtype'] = RefSpec(target_type='AnotherType',
reftype='object')
ret = AttributeSpec.build_spec(spec_dict)
self.assertTrue(isinstance(ret, AttributeSpec))
self.assertDictEqual(ret, expected)
def test_build_spec_no_doc(self):
spec_dict = {'name': 'attribute1', 'dtype': 'text'}
msg = "AttributeSpec.__init__: missing argument 'doc'"
with self.assertRaisesWith(TypeError, msg):
AttributeSpec.build_spec(spec_dict)
