def test_merge_multiple_keys(self):
new_doc = {'identifiers': ['osm:123', 'naptan:xyz'], 'tags': ['bus stop']}
current_doc = {'identifiers': ['foo:678'], 'tags': ['public urinal']}
merged_doc = merge_keys(current_doc, new_doc, ['identifiers', 'tags'])
self.assertEqual(set(['osm:123', 'naptan:xyz', 'foo:678']),
set(merged_doc['identifiers']))
self.assertEqual(set(['bus stop', 'public urinal']),
set(merged_doc['tags']))
def test_merge_keys(self):
new_doc = {'identifiers': ['osm:123', 'naptan:xyz']}
current_doc = {'identifiers': ['foo:678']}
merged_doc = merge_keys(current_doc, new_doc, ['identifiers'])
self.assertEqual(set(['osm:123', 'naptan:xyz', 'foo:678']),
set(merged_doc['identifiers']))
