From de54127472b9d9056a21e1464ba0ad939f99d814 Mon Sep 17 00:00:00 2001
From: relf <remi.lafage@onera.fr>
Date: Wed, 27 Mar 2024 17:55:20 +0100
Subject: [PATCH] Add Gpx derivatives prediction API test

---
 python/egobox/tests/test_gpmix.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/python/egobox/tests/test_gpmix.py b/python/egobox/tests/test_gpmix.py
index 717e6212..770fc7c8 100644
--- a/python/egobox/tests/test_gpmix.py
+++ b/python/egobox/tests/test_gpmix.py
@@ -33,6 +33,12 @@ def test_gpx_kriging(self):
         self.assertAlmostEqual(
             0.0, gpx.predict_var(np.array([[1.1]])).item(), delta=1e-3
         )
+        self.assertAlmostEqual(
+            1.1204, gpx.predict_derivatives(np.array([[1.1]])).item(), delta=1e-3
+        )
+        self.assertAlmostEqual(
+            0.0092, gpx.predict_var_derivatives(np.array([[1.1]])).item(), delta=1e-3
+        )
 
     def test_gpx_save_load(self):
         filename = "gpdump.json"