def test_name_match_selector_backward_compatability(self) -> None:
selector = NameMatchSelector(select_name="pack1")
selector.initialize()
packs = selector.select(self.multi_pack)
doc_ids = ["1"]
for doc_id, pack in zip(doc_ids, packs):
self.assertEqual(doc_id, pack.pack_name)
selector = NameMatchSelector("pack1")
selector.initialize()
packs = selector.select(self.multi_pack)
doc_ids = ["1"]
for doc_id, pack in zip(doc_ids, packs):
self.assertEqual(doc_id, pack.pack_name)
def test_name_match_selector(self) -> None:
selector = NameMatchSelector()
selector.initialize(
configs={"select_name": "pack1"},
)
packs = selector.select(self.multi_pack)
doc_ids = ["1"]
for doc_id, pack in zip(doc_ids, packs):
self.assertEqual(doc_id, pack.pack_name)
# Test reverse selection.
selector.initialize(
configs={"select_name": "pack1", "reverse_selection": True},
)
packs = selector.select(self.multi_pack)
doc_ids = ["2", "Three"]
for doc_id, pack in zip(doc_ids, packs):
self.assertEqual(doc_id, pack.pack_name)