mirror of
https://github.com/deepset-ai/haystack.git
synced 2026-01-07 12:37:27 +00:00
feat: Add support for dates in filters (2.0) (#5823)
* Add support for dates in filters * Add tests * Add release note * Update haystack/preview/utils/filters.py Co-authored-by: ZanSara <sara.zanzottera@deepset.ai> --------- Co-authored-by: ZanSara <sara.zanzottera@deepset.ai>
This commit is contained in:
bogdankostic
GitHub
parent
44f0c468ac
commit
719c1c040c
@ -1,4 +1,5 @@
|
||||
from typing import List, Any, Union, Dict
|
||||
from datetime import datetime
|
||||
|
||||
import numpy as np
|
||||
import pandas as pd
|
||||
@ -73,13 +74,18 @@ def _safe_gt(first: Any, second: Any) -> bool:
|
||||
"""
|
||||
Checks if first is bigger than second.
|
||||
|
||||
Works only for numerical values and dates. Strings, lists, tables and tensors all raise exceptions.
|
||||
Works only for numerical values and dates in ISO format (YYYY-MM-DD). Strings, lists, tables and tensors all raise exceptions.
|
||||
"""
|
||||
if not isinstance(first, GT_TYPES) or not isinstance(second, GT_TYPES):
|
||||
raise FilterError(
|
||||
f"Can't evaluate '{type(first).__name__} > {type(second).__name__}'. "
|
||||
f"Convert these values into one of the following types: {[type_.__name__ for type_ in GT_TYPES]}"
|
||||
)
|
||||
try:
|
||||
first = datetime.fromisoformat(first)
|
||||
second = datetime.fromisoformat(second)
|
||||
except (ValueError, TypeError):
|
||||
raise FilterError(
|
||||
f"Can't evaluate '{type(first).__name__} > {type(second).__name__}'. "
|
||||
f"Convert these values into one of the following types: {[type_.__name__ for type_ in GT_TYPES]} "
|
||||
f"or a datetime string in ISO 8601 format."
|
||||
)
|
||||
return bool(first > second)
|
||||
|
||||
|
||||
|
||||
@ -0,0 +1,4 @@
|
||||
---
|
||||
preview:
|
||||
- |
|
||||
Add support for dates in filters.
|
||||
505
test/preview/utils/test_filters.py
Normal file
505
test/preview/utils/test_filters.py
Normal file
@ -0,0 +1,505 @@
|
||||
import pytest
|
||||
import pandas as pd
|
||||
import numpy as np
|
||||
|
||||
from haystack.preview import Document
|
||||
from haystack.preview.errors import FilterError
|
||||
from haystack.preview.utils.filters import document_matches_filter
|
||||
|
||||
|
||||
class TestFilterUtils:
|
||||
@pytest.mark.unit
|
||||
def test_eq_match(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": "test"}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_eq_no_match(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": "test1"}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_eq_no_match_missing_key(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name1": "test"}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_explicit_eq(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$eq": "test"}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_eq_different_types(self):
|
||||
document = Document(metadata={"name": 1})
|
||||
filter = {"name": "1"}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_eq_dataframes(self):
|
||||
document = Document(metadata={"name": pd.DataFrame({"a": [1, 2, 3]})})
|
||||
filter = {"name": pd.DataFrame({"a": [1, 2, 3]})}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_eq_dataframes_no_match(self):
|
||||
document = Document(metadata={"name": pd.DataFrame({"a": [1, 2, 3]})})
|
||||
filter = {"name": pd.DataFrame({"a": [1, 2, 4]})}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_eq_np_arrays(self):
|
||||
document = Document(metadata={"name": np.array([1, 2, 3])})
|
||||
filter = {"name": np.array([1, 2, 3])}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_eq_np_arrays_no_match(self):
|
||||
document = Document(metadata={"name": np.array([1, 2, 3])})
|
||||
filter = {"name": np.array([1, 2, 4])}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_ne_match(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$ne": "test1"}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_ne_no_match(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$ne": "test"}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_ne_no_match_missing_key(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name1": {"$ne": "test"}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_ne_different_types(self):
|
||||
document = Document(metadata={"name": 1})
|
||||
filter = {"name": {"$ne": "1"}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_ne_dataframes(self):
|
||||
document = Document(metadata={"name": pd.DataFrame({"a": [1, 2, 3]})})
|
||||
filter = {"name": {"$ne": pd.DataFrame({"a": [1, 2, 4]})}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_ne_dataframes_no_match(self):
|
||||
document = Document(metadata={"name": pd.DataFrame({"a": [1, 2, 3]})})
|
||||
filter = {"name": {"$ne": pd.DataFrame({"a": [1, 2, 3]})}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_ne_np_arrays(self):
|
||||
document = Document(metadata={"name": np.array([1, 2, 3])})
|
||||
filter = {"name": {"$ne": np.array([1, 2, 4])}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_ne_np_arrays_no_match(self):
|
||||
document = Document(metadata={"name": np.array([1, 2, 3])})
|
||||
filter = {"name": {"$ne": np.array([1, 2, 3])}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_in_match_list(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$in": ["test", "test1"]}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_in_no_match_list(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$in": ["test2", "test3"]}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_in_implicit(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": ["test", "test1"]}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_in_match_set(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$in": {"test", "test1"}}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_in_no_match_set(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$in": {"test2", "test3"}}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_in_match_tuple(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$in": ("test", "test1")}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_in_no_match_tuple(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$in": ("test2", "test3")}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_in_no_match_missing_key(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name1": {"$in": ["test", "test1"]}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_in_unsupported_type(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$in": "unsupported"}}
|
||||
with pytest.raises(FilterError, match=r"\$in accepts only iterable values like lists, sets and tuples"):
|
||||
document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_nin_match_list(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$nin": ["test1", "test2"]}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_nin_no_match_list(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$nin": ["test", "test1"]}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_nin_match_set(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$nin": {"test1", "test2"}}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_nin_no_match_set(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$nin": {"test", "test1"}}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_nin_match_tuple(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$nin": ("test1", "test2")}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_nin_no_match_tuple(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$nin": ("test", "test1")}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_nin_no_match_missing_key(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name1": {"$nin": ["test", "test1"]}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_nin_unsupported_type(self):
|
||||
document = Document(metadata={"name": "test"})
|
||||
filter = {"name": {"$nin": "unsupported"}}
|
||||
with pytest.raises(FilterError, match=r"\$in accepts only iterable values like lists, sets and tuples"):
|
||||
document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_match_int(self):
|
||||
document = Document(metadata={"age": 21})
|
||||
filter = {"age": {"$gt": 20}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_no_match_int(self):
|
||||
document = Document(metadata={"age": 19})
|
||||
filter = {"age": {"$gt": 20}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_match_float(self):
|
||||
document = Document(metadata={"number": 90.5})
|
||||
filter = {"number": {"$gt": 90.0}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_no_match_float(self):
|
||||
document = Document(metadata={"number": 89.5})
|
||||
filter = {"number": {"$gt": 90.0}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_match_np_number(self):
|
||||
document = Document(metadata={"value": np.float64(7.5)})
|
||||
filter = {"value": {"$gt": np.float64(7.0)}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_no_match_np_number(self):
|
||||
document = Document(metadata={"value": np.float64(6.5)})
|
||||
filter = {"value": {"$gt": np.float64(7.0)}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_match_date_string(self):
|
||||
document = Document(metadata={"date": "2022-01-02"})
|
||||
filter = {"date": {"$gt": "2022-01-01"}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_no_match_date_string(self):
|
||||
document = Document(metadata={"date": "2022-01-01"})
|
||||
filter = {"date": {"$gt": "2022-01-01"}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_no_match_missing_key(self):
|
||||
document = Document(metadata={"age": 21})
|
||||
filter = {"age1": {"$gt": 20}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gt_unsupported_type(self):
|
||||
document = Document(metadata={"age": 21})
|
||||
filter = {"age": {"$gt": "unsupported"}}
|
||||
with pytest.raises(
|
||||
FilterError,
|
||||
match=(
|
||||
r"Convert these values into one of the following types: \['int', 'float', 'number'\] or a datetime string "
|
||||
"in ISO 8601 format"
|
||||
),
|
||||
):
|
||||
document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gte_match_int(self):
|
||||
document = Document(metadata={"age": 21})
|
||||
filter_1 = {"age": {"$gte": 21}}
|
||||
filter_2 = {"age": {"$gte": 20}}
|
||||
assert document_matches_filter(filter_1, document)
|
||||
assert document_matches_filter(filter_2, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gte_no_match_int(self):
|
||||
document = Document(metadata={"age": 20})
|
||||
filter = {"age": {"$gte": 21}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gte_match_float(self):
|
||||
document = Document(metadata={"number": 90.5})
|
||||
filter_1 = {"number": {"$gte": 90.5}}
|
||||
filter_2 = {"number": {"$gte": 90.4}}
|
||||
assert document_matches_filter(filter_1, document)
|
||||
assert document_matches_filter(filter_2, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gte_no_match_float(self):
|
||||
document = Document(metadata={"number": 90.4})
|
||||
filter = {"number": {"$gte": 90.5}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gte_match_np_number(self):
|
||||
document = Document(metadata={"value": np.float64(7.5)})
|
||||
filter_1 = {"value": {"$gte": np.float64(7.5)}}
|
||||
filter_2 = {"value": {"$gte": np.float64(7.4)}}
|
||||
assert document_matches_filter(filter_1, document)
|
||||
assert document_matches_filter(filter_2, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gte_no_match_np_number(self):
|
||||
document = Document(metadata={"value": np.float64(7.4)})
|
||||
filter = {"value": {"$gte": np.float64(7.5)}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gte_match_date_string(self):
|
||||
document = Document(metadata={"date": "2022-01-02"})
|
||||
filter_1 = {"date": {"$gte": "2022-01-02"}}
|
||||
filter_2 = {"date": {"$gte": "2022-01-01"}}
|
||||
assert document_matches_filter(filter_1, document)
|
||||
assert document_matches_filter(filter_2, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gte_no_match_date_string(self):
|
||||
document = Document(metadata={"date": "2022-01-01"})
|
||||
filter = {"date": {"$gte": "2022-01-02"}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_gte_unsupported_type(self):
|
||||
document = Document(metadata={"age": 21})
|
||||
filter = {"age": {"$gte": "unsupported"}}
|
||||
with pytest.raises(
|
||||
FilterError,
|
||||
match=(
|
||||
r"Convert these values into one of the following types: \['int', 'float', 'number'\] or a datetime string "
|
||||
"in ISO 8601 format"
|
||||
),
|
||||
):
|
||||
document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lt_match_int(self):
|
||||
document = Document(metadata={"age": 19})
|
||||
filter = {"age": {"$lt": 20}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lt_no_match_int(self):
|
||||
document = Document(metadata={"age": 20})
|
||||
filter = {"age": {"$lt": 20}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lt_match_float(self):
|
||||
document = Document(metadata={"number": 89.9})
|
||||
filter = {"number": {"$lt": 90.0}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lt_no_match_float(self):
|
||||
document = Document(metadata={"number": 90.0})
|
||||
filter = {"number": {"$lt": 90.0}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lt_match_np_number(self):
|
||||
document = Document(metadata={"value": np.float64(6.9)})
|
||||
filter = {"value": {"$lt": np.float64(7.0)}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lt_no_match_np_number(self):
|
||||
document = Document(metadata={"value": np.float64(7.0)})
|
||||
filter = {"value": {"$lt": np.float64(7.0)}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lt_match_date_string(self):
|
||||
document = Document(metadata={"date": "2022-01-01"})
|
||||
filter = {"date": {"$lt": "2022-01-02"}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lt_no_match_date_string(self):
|
||||
document = Document(metadata={"date": "2022-01-02"})
|
||||
filter = {"date": {"$lt": "2022-01-02"}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lt_unsupported_type(self):
|
||||
document = Document(metadata={"age": 21})
|
||||
filter = {"age": {"$lt": "unsupported"}}
|
||||
with pytest.raises(
|
||||
FilterError,
|
||||
match=(
|
||||
r"Convert these values into one of the following types: \['int', 'float', 'number'\] or a datetime string "
|
||||
"in ISO 8601 format"
|
||||
),
|
||||
):
|
||||
document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lte_match_int(self):
|
||||
document = Document(metadata={"age": 21})
|
||||
filter_1 = {"age": {"$lte": 21}}
|
||||
filter_2 = {"age": {"$lte": 20}}
|
||||
assert not document_matches_filter(filter_2, document)
|
||||
assert document_matches_filter(filter_1, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lte_no_match_int(self):
|
||||
document = Document(metadata={"age": 22})
|
||||
filter = {"age": {"$lte": 21}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lte_match_float(self):
|
||||
document = Document(metadata={"number": 90.5})
|
||||
filter_1 = {"number": {"$lte": 90.5}}
|
||||
filter_2 = {"number": {"$lte": 90.4}}
|
||||
assert not document_matches_filter(filter_2, document)
|
||||
assert document_matches_filter(filter_1, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lte_no_match_float(self):
|
||||
document = Document(metadata={"number": 90.6})
|
||||
filter = {"number": {"$lte": 90.5}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lte_match_np_number(self):
|
||||
document = Document(metadata={"value": np.float64(7.5)})
|
||||
filter_1 = {"value": {"$lte": np.float64(7.5)}}
|
||||
filter_2 = {"value": {"$lte": np.float64(7.4)}}
|
||||
assert not document_matches_filter(filter_2, document)
|
||||
assert document_matches_filter(filter_1, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lte_no_match_np_number(self):
|
||||
document = Document(metadata={"value": np.float64(7.6)})
|
||||
filter = {"value": {"$lte": np.float64(7.5)}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lte_match_date_string(self):
|
||||
document = Document(metadata={"date": "2022-01-02"})
|
||||
filter_1 = {"date": {"$lte": "2022-01-02"}}
|
||||
filter_2 = {"date": {"$lte": "2022-01-01"}}
|
||||
assert not document_matches_filter(filter_2, document)
|
||||
assert document_matches_filter(filter_1, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lte_no_match_date_string(self):
|
||||
document = Document(metadata={"date": "2022-01-03"})
|
||||
filter = {"date": {"$lte": "2022-01-02"}}
|
||||
assert not document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_lte_unsupported_type(self):
|
||||
document = Document(metadata={"age": 21})
|
||||
filter = {"age": {"$lte": "unsupported"}}
|
||||
with pytest.raises(
|
||||
FilterError,
|
||||
match=(
|
||||
r"Convert these values into one of the following types: \['int', 'float', 'number'\] or a datetime string "
|
||||
"in ISO 8601 format"
|
||||
),
|
||||
):
|
||||
document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_implicit_and(self):
|
||||
document = Document(metadata={"age": 21, "name": "John"})
|
||||
filter = {"age": {"$gt": 18}, "name": "John"}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_explicit_and(self):
|
||||
document = Document(metadata={"age": 21})
|
||||
filter = {"age": {"$and": {"$gt": 18}, "$lt": 25}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_or(self):
|
||||
document = Document(metadata={"age": 26})
|
||||
filter = {"age": {"$or": [{"$gt": 18}, {"$lt": 25}]}}
|
||||
assert document_matches_filter(filter, document)
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_not(self):
|
||||
document = Document(metadata={"age": 17})
|
||||
filter = {"age": {"$not": {"$gt": 18}}}
|
||||
assert document_matches_filter(filter, document)
|
||||
Loading…
x
Reference in New Issue
Block a user