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]}}