From decd3b8b262fd740fb6d53b0d976f4c58d109b06 Mon Sep 17 00:00:00 2001
From: estelle <estelle.scifo@neo4j.com>
Date: Mon, 6 Jan 2025 12:14:35 +0100
Subject: [PATCH] Ruff

---
 tests/unit/utils/test_prettifyer.py | 20 +++++++++-----------
 1 file changed, 9 insertions(+), 11 deletions(-)

diff --git a/tests/unit/utils/test_prettifyer.py b/tests/unit/utils/test_prettifyer.py
index badf5827..a5b13850 100644
--- a/tests/unit/utils/test_prettifyer.py
+++ b/tests/unit/utils/test_prettifyer.py
@@ -13,7 +13,7 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-from unittest.mock import call, patch, Mock
+from unittest.mock import Mock, call, patch
 
 from neo4j_graphrag.utils.logging import Prettifyer, prettify
 
@@ -101,20 +101,18 @@ def test_prettifyer_dict_nested() -> None:
 
 
 def test_prettify_function() -> None:
-    assert prettify({
-        "key": {
-            "key0.1": "ab" * 200,
-            "key0.2": ["a"] * 10,
-            "key0.3": {
-                "key0.3.1": "a short strng"
+    assert prettify(
+        {
+            "key": {
+                "key0.1": "ab" * 200,
+                "key0.2": ["a"] * 10,
+                "key0.3": {"key0.3.1": "a short strng"},
             }
         }
-    }) == {
+    ) == {
         "key": {
             "key0.1": "ab" * 100 + f"... (200 chars)",
             "key0.2": ["a"] * 5 + ["... (5 items)"],
-            "key0.3": {
-                "key0.3.1": "a short strng"
-            }
+            "key0.3": {"key0.3.1": "a short strng"},
         }
     }