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