tests: updating processor test

Loop3D · Nov 18, 2024 · 4a25702 · 4a25702
1 parent e508d2d
commit 4a25702
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/input/test_data_processor.py b/tests/unit/input/test_data_processor.py
@@ -7,7 +7,7 @@
 def test_create_processor():
     df = pd.DataFrame(rng.random(size=(10, 3)), columns=["X", "Y", "Z"])
     df["name"] = ["unit_{}".format(name % 2) for name in range(10)]
-    stratigraphic_order = [("sg", ["unit_0", "unit_1"])]
+    stratigraphic_order = [("sg", ["unit_0", "unit_1", "basement"])]
     thicknesses = {"unit_0": 1.0, "unit_1": 0.5}
     processor = ProcessInputData(
         contacts=df, stratigraphic_order=stratigraphic_order, thicknesses=thicknesses