Python Config.from_env Examples

Example #1

File: test_keras_model_to_tf_model.py Project: jiangjiajun/Adlik

    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')]))

Example #2

File: test_keras_model_to_tf_model.py Project: tmaone/Adlik

    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))

Example #3

    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)
                   ]))

Example #4

File: test_keras_model_to_tf_model.py Project: tmaone/Adlik

    def test_from_env_minimal(self):
        config = Config.from_env({})

        self.assertEqual(config, Config(input_nodes=None, output_nodes=None))