mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-09-22 14:43:07 +00:00
7287657f0e
* Rework Document serialisation Make Document backward compatible Fix InMemoryDocumentStore filters Fix InMemoryDocumentStore.bm25_retrieval Add release notes Fix pylint failures Enhance Document kwargs handling and docstrings Rename Document's text field to content Fix e2e tests Fix SimilarityRanker tests Fix typo in release notes Rename Document's metadata field to meta (#6183) * fix bugs * make linters happy * fix * more fix * match regex --------- Co-authored-by: Massimiliano Pippi <mpippi@gmail.com>
24 lines
910 B
Python
24 lines
910 B
Python
import pytest
|
|
|
|
from haystack.preview import Document
|
|
from haystack.preview.components.routers.metadata_router import MetadataRouter
|
|
|
|
|
|
class TestMetadataRouter:
|
|
@pytest.mark.unit
|
|
def test_run(self):
|
|
rules = {
|
|
"edge_1": {"created_at": {"$gte": "2023-01-01", "$lt": "2023-04-01"}},
|
|
"edge_2": {"created_at": {"$gte": "2023-04-01", "$lt": "2023-07-01"}},
|
|
}
|
|
router = MetadataRouter(rules=rules)
|
|
documents = [
|
|
Document(meta={"created_at": "2023-02-01"}),
|
|
Document(meta={"created_at": "2023-05-01"}),
|
|
Document(meta={"created_at": "2023-08-01"}),
|
|
]
|
|
output = router.run(documents=documents)
|
|
assert output["edge_1"][0].meta["created_at"] == "2023-02-01"
|
|
assert output["edge_2"][0].meta["created_at"] == "2023-05-01"
|
|
assert output["unmatched"][0].meta["created_at"] == "2023-08-01"
|