From df396240d97adc6e3a8e69a6ce3d994b1d576c3e Mon Sep 17 00:00:00 2001
From: akhynkokateryna <kateryna.akhynko@ucu.edu.ua>
Date: Tue, 3 Dec 2024 20:31:07 +0200
Subject: [PATCH] correct test

---
 sharp/visualization/tests/test_visualization.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/sharp/visualization/tests/test_visualization.py b/sharp/visualization/tests/test_visualization.py
index 50ab7fa..108f892 100644
--- a/sharp/visualization/tests/test_visualization.py
+++ b/sharp/visualization/tests/test_visualization.py
@@ -13,8 +13,9 @@ def mock_sharp():
 
     class MockSharp:
         def __init__(self):
-            self.qoi = "rank"
-            self.measure = "shapley"
+            self.feature_names_ = np.array(["Feature1", "Feature2", "Feature3"])
+            self.qoi="rank"
+            self.measure="shapley"
 
     return MockSharp()