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() == {"missing": {"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]}}
def __le__(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 LessEqual(field=self.name, value=other)
else:
raise NotImplementedError(other, type(other))
def __ge__(self, other):
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 GreaterEqual(field=self.name, value=other)
else:
raise NotImplementedError(other, type(other))