

def test_filter_doc_ids_by_doc_type(self):
    expected_doc_ids = {doc['_id'] for doc in self.docs}
    input_doc_ids = (expected_doc_ids ^ {'other', 'random', 'ids'})
    actual_doc_ids = set(filter_doc_ids_by_doc_type(self.migration.source_db, input_doc_ids, self.migration.doc_types))
    self.assertEqual(expected_doc_ids, actual_doc_ids)

dataset/ETHPy150Open dimagi/commcare-hq/corehq/doctypemigrations/tests.py TestDocTypeMigrations.test_filter_doc_ids_by_doc_type/original