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