def test_from_env_full(self):
config = Config.from_env({'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_env_input_names_only(self):
config = Config.from_env({'input_layer_names': 'abc,def:3'})
self.assertEqual(
config,
Config(input_nodes=[
NodeSpec(layer_name='abc'),
NodeSpec(layer_name='def', node_index=3)
],
output_nodes=None))
def test_from_env_output_names_only(self):
config = Config.from_env({'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_env_minimal(self):
config = Config.from_env({})
self.assertEqual(config, Config(input_nodes=None, output_nodes=None))