Format expected json

MetaphorData · Oct 1, 2024 · 30a7114 · 30a7114
1 parent f77be1e
commit 30a7114
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/mongodb/test_extractor.py b/tests/mongodb/test_extractor.py
@@ -84,4 +84,4 @@ def get_dataset_name(dataset: Dataset):
 
     trimmed_datasets = [EventUtil.trim_event(ds) for ds in datasets]
     expected = f"{test_root_dir}/mongodb/expected_datasets.json"
-    assert trimmed_datasets == load_json(expected, trimmed_datasets)
+    assert trimmed_datasets == load_json(expected)
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -16,7 +16,8 @@ def load_json(path, events: Optional[List[Dict[str, Any]]] = None):
     """
     if events:
         with open(path, "w") as f:
-            f.write(json.dumps(events, indent=4))
+            f.write(json.dumps(events, indent=2, sort_keys=True))
+            f.write("\n")
     with open(path, "r") as f:
         return json.load(f)