def test_regex_name_match_selector(self) -> None:
selector = RegexNameMatchSelector()
selector.initialize(
configs={"select_name": "^.*\\d$"},
)
packs = selector.select(self.multi_pack)
doc_ids = ["1", "2"]
for doc_id, pack in zip(doc_ids, packs):
self.assertEqual(doc_id, pack.pack_name)
# Test reverse selection.
selector.initialize(
{"select_name": "^.*\\d$", "reverse_selection": True}
)
packs = selector.select(self.multi_pack)
doc_ids = ["Three"]
for doc_id, pack in zip(doc_ids, packs):
self.assertEqual(doc_id, pack.pack_name)
def test_regex_name_match_selector_backward_compatability(self) -> None:
selector = RegexNameMatchSelector(select_name="^.*\\d$")
selector.initialize()
packs = selector.select(self.multi_pack)
doc_ids = ["1", "2"]
for doc_id, pack in zip(doc_ids, packs):
self.assertEqual(doc_id, pack.pack_name)
# Test different configuration method (backward compatibility)
selector = RegexNameMatchSelector("^.*\\d$")
selector.initialize()
packs = selector.select(self.multi_pack)
doc_ids = ["1", "2"]
for doc_id, pack in zip(doc_ids, packs):
self.assertEqual(doc_id, pack.pack_name)
# Test reverse selection.
selector.initialize({"reverse_selection": True})
packs = selector.select(self.multi_pack)
doc_ids = ["Three"]
for doc_id, pack in zip(doc_ids, packs):
self.assertEqual(doc_id, pack.pack_name)