fix imports and formatting

1 files changed, 14 insertions, 3 deletions

diff --git a/tests/test_utils.py b/tests/test_utils.py
index 24be9d1..381c44e 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -99,11 +99,22 @@ def test_zstdlines():
         with open(fn) as f:
             assert [s.strip() for s in f.readlines()] == list(zstdlines(zfn))
 
+
 def test_es_compat_hits_total():
     cases = (
-        ({"hits": {"total": 6}}, 6),
-        ({"hits": {"total": {"value": 7, "relation": "eq"}}}, 7),
+        ({
+            "hits": {
+                "total": 6
+            }
+        }, 6),
+        ({
+            "hits": {
+                "total": {
+                    "value": 7,
+                    "relation": "eq"
+                }
+            }
+        }, 7),
     )
     for r, expected in cases:
         assert es_compat_hits_total(r) == expected
-