def test_stored_data_parameters_format(self):
with self.assertRaises(InvalidFormat):
validate_data_structure({
'layers': [{
'name': 'name',
'parameters': {
'weight': np.ones((2, 3)),
}
}]
})
with self.assertRaises(InvalidFormat):
validate_data_structure({
'layers': [{
'name': 'name',
'parameters': {
'weight': {
'data': np.ones((2, 3)),
},
}
}]
})
result = validate_data_structure({
'layers': [{
'name': 'name',
'parameters': {
'weight': {
'value': np.ones((2, 3)),
'trainable': True,
},
}
}]
})
self.assertIsNone(result)
def test_stored_data_dict_format_basics(self):
with self.assertRaises(InvalidFormat):
validate_data_structure([])
with self.assertRaises(InvalidFormat):
validate_data_structure({})
with self.assertRaises(InvalidFormat):
validate_data_structure({'layers': {}})
with self.assertRaises(InvalidFormat):
validate_data_structure({'layers': []})
def test_stored_data_layers_format(self):
with self.assertRaises(InvalidFormat):
validate_data_structure({'layers': [[]]})
with self.assertRaises(InvalidFormat):
validate_data_structure({'layers': [{
'parameters': {},
}]})
with self.assertRaises(InvalidFormat):
validate_data_structure({'layers': [{
'parameters': {},
}]})
with self.assertRaises(InvalidFormat):
validate_data_structure({'layers': [{
'parameters': {},
}]})
with self.assertRaises(InvalidFormat):
validate_data_structure({
'layers': [{
'parameters': [], # wrong type
'name': 'name',
}]
})
result = validate_data_structure(
{'layers': [{
'parameters': {},
'name': 'name',
}]})
self.assertIsNone(result)