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)