Add MatrixStats aggregation (#1631) (#1798)

(cherry picked from commit 3564610) Co-authored-by: Hüseyin Emre Armağan <[email protected]>
elastic · Apr 29, 2024 · 74c5f7b · 74c5f7b
1 parent 6583c22
commit 74c5f7b
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])