[Backport 8.x] Added "adjacency_matrix" and "top_metrics" aggregations

elasticsearch_dsl/aggs.py

@@ -197,6 +197,10 @@ class AutoDateHistogram(DateHistogram):
     name = "auto_date_histogram"
 
 
+class AdjacencyMatrix(Bucket):
+    name = "adjacency_matrix"
+
+
 class DateRange(Bucket):
     name = "date_range"
 
@@ -385,6 +389,10 @@ class Sum(Agg):
     name = "sum"
 
 
+class TopMetrics(Agg):
+    name = "top_metrics"
+
+
 class TTest(Agg):
     name = "t_test"
 

tests/test_aggs.py

@@ -449,3 +449,30 @@ def test_random_sampler_aggregation():
             },
         },
     } == a.to_dict()
+
+
+def test_adjancecy_matrix_aggregation():
+    a = aggs.AdjacencyMatrix(
+        filters={
+            "grpA": {"terms": {"accounts": ["hillary", "sidney"]}},
+            "grpB": {"terms": {"accounts": ["donald", "mitt"]}},
+            "grpC": {"terms": {"accounts": ["vladimir", "nigel"]}},
+        }
+    )
+    assert {
+        "adjacency_matrix": {
+            "filters": {
+                "grpA": {"terms": {"accounts": ["hillary", "sidney"]}},
+                "grpB": {"terms": {"accounts": ["donald", "mitt"]}},
+                "grpC": {"terms": {"accounts": ["vladimir", "nigel"]}},
+            }
+        }
+    } == a.to_dict()
+
+
+def test_top_metrics_aggregation():
+    a = aggs.TopMetrics(metrics={"field": "m"}, sort={"s": "desc"})
+
+    assert {
+        "top_metrics": {"metrics": {"field": "m"}, "sort": {"s": "desc"}}
+    } == a.to_dict()