def mock_lookup(lookup_input, lookup_type, raw=None): if raw is None: raw = "%s %s" % (lookup_type, lookup_input) return Lookup(type=lookup_type, input=lookup_input, raw=raw)
def mock_lookup(lookup_input, lookup_type='output', raw=None): if raw is None: raw = lookup_input return Lookup(type=lookup_type, input=lookup_input, raw=raw)