def __lt__(self, other: Union[int, float, "Series"]) -> BooleanFilter:
if isinstance(other, Series):
# Need to use scripted query to compare to values
painless = f"doc['{self.name}'].value < doc['{other.name}'].value"
return ScriptFilter(painless, lang="painless")
elif isinstance(other, (int, float)):
return Less(field=self.name, value=other)
else:
raise NotImplementedError(other, type(other))
def test_and_filter3(self):
exp = GreaterEqual("a", 2) & (Less("b", 3) & Equal("c", 4))
assert exp.build() == {
"bool": {
"must": [
{"range": {"b": {"lt": 3}}},
{"term": {"c": 4}},
{"range": {"a": {"gte": 2}}},
]
}
}
def test_or_filter2(self):
exp = GreaterEqual("a", 2) | Less("b", 3) | Equal("c", 4)
assert exp.build() == {
"bool": {
"should": [
{"range": {"a": {"gte": 2}}},
{"range": {"b": {"lt": 3}}},
{"term": {"c": 4}},
]
}
}
def test_not_and_filter(self):
exp = ~(GreaterEqual("a", 2) & Less("b", 3))
assert exp.build() == {
"bool": {
"must_not": {
"bool": {
"must": [
{"range": {"a": {"gte": 2}}},
{"range": {"b": {"lt": 3}}},
]
}
}
}
}
def test_leaf_boolean_filter(self):
assert GreaterEqual("a", 2).build() == {"range": {"a": {"gte": 2}}}
assert LessEqual("a", 2).build() == {"range": {"a": {"lte": 2}}}
assert Less("a", 2).build() == {"range": {"a": {"lt": 2}}}
assert Equal("a", 2).build() == {"term": {"a": 2}}
exp = Equal("a", 2)
assert (~exp).build()["bool"], {"must_not": {"term": {"a": 2}}}
assert Greater("a", 2).build() == {"range": {"a": {"gt": 2}}}
assert IsIn("a", [1, 2, 3]).build() == {"terms": {"a": [1, 2, 3]}}
assert Like("a", "a*b").build() == {"wildcard": {"a": "a*b"}}
assert Rlike("a", "a*b").build() == {"regexp": {"a": "a*b"}}
assert Startswith("a", "jj").build() == {"prefix": {"a": "jj"}}
assert IsNull("a").build() == {
"bool": {
"must_not": {
"exists": {
"field": "a"
}
}
}
}
assert NotNull("a").build() == {"exists": {"field": "a"}}
assert ScriptFilter('doc["num1"].value > params.param1',
lang="painless",
params={
"param1": 5
}).build() == {
"script": {
"script": {
"lang": "painless",
"source":
'doc["num1"].value > params.param1',
"params": {
"param1": 5
},
}
}
}
assert IsIn("ids", [1, 2, 3]).build() == {"ids": {"values": [1, 2, 3]}}
