Python BucketSort Examples

Example #1

File: test_aggs.py Project: elastic/elasticsearch-dsl-py

def test_bucket_sort_agg_only_trnunc():
    bucket_sort_agg = aggs.BucketSort(**{"from": 1, "size": 1})
    assert bucket_sort_agg.to_dict() == {"bucket_sort": {"from": 1, "size": 1}}

    a = aggs.DateHistogram(field="date", interval="month")
    a.bucket("bucket_truncate", "bucket_sort", **{"from": 1, "size": 1})
    assert {
        "date_histogram": {"field": "date", "interval": "month"},
        "aggs": {"bucket_truncate": {"bucket_sort": {"from": 1, "size": 1}}},
    } == a.to_dict()

Example #2

File: test_aggs.py Project: olcaycoban/elasticsearch-dsl-py

def test_bucket_sort_agg():
    bucket_sort_agg = aggs.BucketSort(sort=[{
        "total_sales": {
            "order": "desc"
        }
    }],
                                      size=3)
    assert bucket_sort_agg.to_dict() == {
        "bucket_sort": {
            "sort": [{
                "total_sales": {
                    "order": "desc"
                }
            }],
            "size": 3
        }
    }

    a = aggs.DateHistogram(field="date", interval="month")
    a.bucket("total_sales", "sum", field="price")
    a.bucket(
        "sales_bucket_sort",
        "bucket_sort",
        sort=[{
            "total_sales": {
                "order": "desc"
            }
        }],
        size=3,
    )
    assert {
        "date_histogram": {
            "field": "date",
            "interval": "month"
        },
        "aggs": {
            "total_sales": {
                "sum": {
                    "field": "price"
                }
            },
            "sales_bucket_sort": {
                "bucket_sort": {
                    "sort": [{
                        "total_sales": {
                            "order": "desc"
                        }
                    }],
                    "size": 3
                }
            },
        },
    } == a.to_dict()

Example #3

def test_bucket_sort_agg():
    bucket_sort_agg = aggs.BucketSort(sort=[{
        "total_sales": {
            "order": "desc"
        }
    }],
                                      size=3)
    assert bucket_sort_agg.to_dict() == {
        "bucket_sort": {
            "sort": [{
                "total_sales": {
                    "order": "desc"
                }
            }],
            "size": 3
        }
    }

    a = aggs.DateHistogram(field='date', interval='month')
    a.bucket('total_sales', 'sum', field='price')
    a.bucket('sales_bucket_sort',
             'bucket_sort',
             sort=[{
                 "total_sales": {
                     "order": "desc"
                 }
             }],
             size=3)
    assert {
        "date_histogram": {
            "field": "date",
            "interval": "month"
        },
        "aggs": {
            "total_sales": {
                "sum": {
                    "field": "price"
                }
            },
            "sales_bucket_sort": {
                "bucket_sort": {
                    "sort": [{
                        "total_sales": {
                            "order": "desc"
                        }
                    }],
                    "size": 3
                }
            }
        }
    } == a.to_dict()

Example #4

def test_bucket_sort_agg_only_trnunc():
    bucket_sort_agg = aggs.BucketSort(**{'from': 1, 'size': 1})
    assert bucket_sort_agg.to_dict() == {"bucket_sort": {"from": 1, "size": 1}}

    a = aggs.DateHistogram(field='date', interval='month')
    a.bucket('bucket_truncate', 'bucket_sort', **{'from': 1, 'size': 1})
    assert {
        "date_histogram": {
            "field": "date",
            "interval": "month"
        },
        "aggs": {
            "bucket_truncate": {
                "bucket_sort": {
                    "from": 1,
                    "size": 1
                }
            }
        }
    } == a.to_dict()