Python get_key_to_references_mapping Examples

Example #1

File: test_view_references.py Project: nagyistoce/dnn-brainstorm

def test_get_key_to_references_mapping_default():
    keys = ['InputLayer', 'ForwardLayer_1', 'ForwardLayer_2']
    references = ['*Layer_2', 'default']
    key_to_refs = get_key_to_references_mapping(keys, references)
    assert key_to_refs == {
        'InputLayer': {'default'},
        'ForwardLayer_1': {'default'},
        'ForwardLayer_2': {'*Layer_2'}
    }

Example #2

File: test_view_references.py Project: nagyistoce/dnn-brainstorm

def test_get_key_to_references_mapping_raises_non_matching_ref():
    keys = ['InputLayer', 'ForwardLayer_1', 'ForwardLayer_2']
    references = ['*Layer_2', 'Lstm*']
    with pytest.raises(NetworkValidationError):
        key_to_refs = get_key_to_references_mapping(keys, references)