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 __ne__(self, other: Union[int, float, str, "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 NotFilter(Equal(field=self.name, value=other))
elif isinstance(other, str):
return NotFilter(Equal(field=self.name, value=other))
else:
raise NotImplementedError(other, type(other))
def __eq__(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)
elif isinstance(other, (int, float)):
return Equal(field=self.name, value=other)
elif isinstance(other, str):
return Equal(field=self.name, value=other)
else:
raise NotImplementedError(other, type(other))
def test_complex_filter(self):
exp = (Greater("a", 10)
& ~Greater("a", 250)
& (Equal("b", "X") | (Equal("b", "Y")))
& (Equal("b", "Z")))
assert exp.build() == {
"bool": {
"must": [
{
"range": {
"a": {
"gt": 10
}
}
},
{
"bool": {
"must_not": {
"range": {
"a": {
"gt": 250
}
}
}
}
},
{
"bool": {
"should": [{
"term": {
"b": "X"
}
}, {
"term": {
"b": "Y"
}
}]
}
},
{
"term": {
"b": "Z"
}
},
]
}
}
def test_and_filter2(self):
exp = GreaterEqual("a", 2) & Less("b", 3) & Equal("c", 4)
print(exp.build())
assert exp.build() == {
"bool": {
"must": [
{
"range": {
"a": {
"gte": 2
}
}
},
{
"range": {
"b": {
"lt": 3
}
}
},
{
"term": {
"c": 4
}
},
]
}
}
def test_and_or_filter(self):
exp = GreaterEqual("a", 2) & (Less("b", 3) | Equal("c", 4))
assert exp.build() == {
"bool": {
"must": [
{
"range": {
"a": {
"gte": 2
}
}
},
{
"bool": {
"should": [{
"range": {
"b": {
"lt": 3
}
}
}, {
"term": {
"c": 4
}
}]
}
},
]
}
}