Add MatrixStats aggregation (#1631)

elastic · Apr 15, 2024 · 3564610 · 3564610
1 parent 6cfbbe1
commit 3564610
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/elasticsearch_dsl/aggs.py b/elasticsearch_dsl/aggs.py
@@ -333,6 +333,10 @@ class Max(Agg):
     name = "max"
 
 
+class MatrixStats(Agg):
+    name = "matrix_stats"
+
+
 class MedianAbsoluteDeviation(Agg):
     name = "median_absolute_deviation"
 

diff --git a/tests/test_aggs.py b/tests/test_aggs.py
@@ -341,6 +341,12 @@ def test_inference_aggregation():
     } == a.to_dict()
 
 
+def test_matrix_stats_aggregation():
+    a = aggs.MatrixStats(fields=["poverty", "income"])
+
+    assert {"matrix_stats": {"fields": ["poverty", "income"]}} == a.to_dict()
+
+
 def test_moving_percentiles_aggregation():
     a = aggs.DateHistogram()
     a.bucket("the_percentile", "percentiles", field="price", percents=[1.0, 99.0])