tests: updating horizontal layer test

Author: lachlangrose

tests/integration/test_interpolator.py

@@ -108,19 +108,11 @@ def test_horizontal_layers(interpolatortype, nelements):
     model = GeologicalModel(bb[0, :], bb[1, :])
 
     model.data = data
-    model.create_and_add_foliation("strati", interpolatortype=interpolatortype, nelements=1e4)
+    model.create_and_add_foliation("strati", interpolatortype=interpolatortype, nelements=nelements)
 
-    assert np.all(np.isclose(model["strati"].evaluate_value(data[["X", "Y", "Z"]]), data["val"]))
-
-
-def test_horizontal_layers(interpolatortype, nelements):
-    data, bb = load_horizontal()
-    model = GeologicalModel(bb[0, :], bb[1, :])
-
-    model.data = data
-    model.create_and_add_foliation("strati", interpolatortype=interpolatortype, nelements=1e4)
-
-    assert np.all(np.isclose(model["strati"].evaluate_value(data[["X", "Y", "Z"]]), data["val"]))
+    assert np.all(
+        np.isclose(model["strati"].evaluate_value(data[["X", "Y", "Z"]]), data["val"], atol=1e-3)
+    )
 
 
 if __name__ == "__main__":
@@ -134,8 +126,8 @@ def test_horizontal_layers(interpolatortype, nelements):
     test_create_stratigraphy_PLI_lu()
     test_create_stratigraphy_PLI_pyamg()
     test_model_with_data_outside_of_bounding_box()
-    test_horizontal_layers("FDI", 1000)
-    test_horizontal_layers("PLI", 1000)
-    test_horizontal_layers("FDI", 1000)
-    test_horizontal_layers("PLI", 1000)
+    test_horizontal_layers("FDI", 10000)
+    test_horizontal_layers("PLI", 10000)
+    test_horizontal_layers("FDI", 10000)
+    test_horizontal_layers("PLI", 10000)
     print("ok")