def test_from_json_full(self):
config = Config.from_json({'input_layer_names': ['abc', 'def:3'],
'output_layer_names': ['ghi:2', 'jkl']})
self.assertEqual(config, Config(input_nodes=[NodeSpec(layer_name='abc'),
NodeSpec(layer_name='def', node_index=3)],
output_nodes=[NodeSpec(layer_name='ghi', node_index=2),
NodeSpec(layer_name='jkl')]))
def test_from_json_output_names_only(self):
config = Config.from_json({'output_layer_names': ['abc', 'def:3']})
self.assertEqual(
config,
Config(input_nodes=None,
output_nodes=[
NodeSpec(layer_name='abc'),
NodeSpec(layer_name='def', node_index=3)
]))
def test_from_json_minimal(self):
config = Config.from_json({})
self.assertEqual(config, Config(input_nodes=None, output_nodes=None))