def test_invert_dict_multi_valued(): d = {'foo': 'bar', 'baz': 'bar'} res = invert_dict(d) assert res == {'bar': set(['foo', 'baz'])}
def test_invert_dict_single_valued(): d = {'foo': 'bar', 'baz': 'qux'} res = invert_dict(d) assert res == {'bar': set(['foo']), 'qux': set(['baz'])}