clean up max file tests

CoffeaTeam · Jan 8, 2024 · e4a2fd6 · e4a2fd6
1 parent df364d3
commit e4a2fd6
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/tests/test_dataset_tools.py b/tests/test_dataset_tools.py
@@ -295,7 +295,9 @@ def test_max_files():
                     "steps": [[0, 7], [7, 14], [14, 21], [21, 28], [28, 35], [35, 40]],
                     "uuid": "a9490124-3648-11ea-89e9-f5b55c90beef",
                 }
-            }
+            },
+            "metadata": None,
+            "form": None,
         },
         "Data": {
             "files": {
@@ -304,7 +306,9 @@ def test_max_files():
                     "steps": [[0, 7], [7, 14], [14, 21], [21, 28], [28, 35], [35, 40]],
                     "uuid": "a210a3f8-3648-11ea-a29f-f5b55c90beef",
                 }
-            }
+            },
+            "metadata": None,
+            "form": None,
         },
     }
 
@@ -313,15 +317,17 @@ def test_slice_files():
     sliced_files = slice_files(_updated_result, slice(1, None, 2))
 
     assert sliced_files == {
-        "ZJets": {"files": {}},
+        "ZJets": {"files": {}, "metadata": None, "form": None},
         "Data": {
             "files": {
                 "tests/samples/nano_dimuon_not_there.root": {
                     "object_path": "Events",
                     "steps": None,
                     "uuid": None,
                 }
-            }
+            },
+            "metadata": None,
+            "form": None,
         },
     }