Rework well trajectory tests

tests/jobs/well_trajectory/test_read_trajectories.py

@@ -1,4 +1,3 @@
-import shutil
 from pathlib import Path
 
 import numpy as np
@@ -131,7 +130,7 @@ def test_read_trajectories_no_kickoff(copy_testdata_tmpdir):
 
 
 def test_read_laterals_orphaned_branches(copy_testdata_tmpdir):
-    copy_testdata_tmpdir(Path(TEST_DATA) / "spe1case1_mlt")
+    copy_testdata_tmpdir(Path(TEST_DATA) / "read_laterals")
     config = ConfigSchema.model_validate(load_yaml("config.yml"))
     mlt_p1_z = load_json("mlt_p1_z.json")
     mlt_p1_z["DUMMY"] = {"1": 0.0}
@@ -143,7 +142,7 @@ def test_read_laterals_orphaned_branches(copy_testdata_tmpdir):
 
 
 def test_read_laterals_missing_files(copy_testdata_tmpdir):
-    copy_testdata_tmpdir(Path(TEST_DATA) / "spe1case1_mlt")
+    copy_testdata_tmpdir(Path(TEST_DATA) / "read_laterals")
     config = ConfigSchema.model_validate(load_yaml("config.yml"))
     Path("mlt_p2_b.json").unlink()
     Path("mlt_p3_x.json").unlink()
@@ -155,7 +154,7 @@ def test_read_laterals_missing_files(copy_testdata_tmpdir):
 
 
 def test_read_laterals_missing_wells(copy_testdata_tmpdir):
-    copy_testdata_tmpdir(Path(TEST_DATA) / "spe1case1_mlt")
+    copy_testdata_tmpdir(Path(TEST_DATA) / "read_laterals")
     config = ConfigSchema.model_validate(load_yaml("config.yml"))
     mlt_p2_b = load_json("mlt_p2_b.json")
     mlt_p3_x = load_json("mlt_p3_x.json")
@@ -171,7 +170,7 @@ def test_read_laterals_missing_wells(copy_testdata_tmpdir):
 
 
 def test_read_laterals_missing_branches(copy_testdata_tmpdir):
-    copy_testdata_tmpdir(Path(TEST_DATA) / "spe1case1_mlt")
+    copy_testdata_tmpdir(Path(TEST_DATA) / "read_laterals")
     config = ConfigSchema.model_validate(load_yaml("config.yml"))
     mlt_p2_b = load_json("mlt_p2_b.json")
     mlt_p3_x = load_json("mlt_p3_x.json")
@@ -187,8 +186,7 @@ def test_read_laterals_missing_branches(copy_testdata_tmpdir):
 
 
 def test_read_laterals_branch_not_on_well(copy_testdata_tmpdir):
-    copy_testdata_tmpdir(Path(TEST_DATA) / "spe1case1_mlt")
-    shutil.copytree("expected/wellpaths", "wellpaths")
+    copy_testdata_tmpdir(Path(TEST_DATA) / "read_laterals")
     config = ConfigSchema.model_validate(load_yaml("config.yml"))
     mlt_p1_z = load_json("mlt_p1_z.json")
     mlt_p1_z["INJ"]["1"] = 1.0
@@ -197,15 +195,8 @@ def test_read_laterals_branch_not_on_well(copy_testdata_tmpdir):
         read_laterals(config.scales, config.references, config.wells)
 
 
-def test_read_laterals_no_branches(copy_testdata_tmpdir):
-    copy_testdata_tmpdir(Path(TEST_DATA) / "spe1case1")
-    config = ConfigSchema.model_validate(load_yaml("config.yml"))
-    assert read_laterals(config.scales, config.references, config.wells) == {}
-
-
 def test_read_laterals(copy_testdata_tmpdir):
-    copy_testdata_tmpdir(Path(TEST_DATA) / "spe1case1_mlt")
-    shutil.copytree("expected/wellpaths", "wellpaths")
+    copy_testdata_tmpdir(Path(TEST_DATA) / "read_laterals")
     config = ConfigSchema.model_validate(load_yaml("config.yml"))
     laterals = read_laterals(config.scales, config.references, config.wells)
 

tests/testdata/well_trajectory/read_laterals/config.yml

@@ -0,0 +1,35 @@
+scales:
+    x: 10000
+    y: 10000
+    z: 100
+    k: 50
+references:
+    x: 5000
+    y: 5000
+    z: 8375
+    k: 50
+interpolation:
+    type: resinsight
+    measured_depth_step: 5.0
+wells:
+    - name: INJ
+      group: G1
+      phase: GAS
+      platform: INJ
+      cost: 0
+      dogleg: 4
+    - name: PROD
+      group: G1
+      phase: OIL
+      platform: PROD
+      cost: 0
+      dogleg: 4
+platforms:
+    - name: INJ
+      x: 2500
+      y: 2500
+      k: 50
+    - name: PROD
+      x: 7500
+      y: 7500
+      k: 50

tests/testdata/well_trajectory/read_laterals/mlt_p1_z.json

@@ -0,0 +1 @@
+{"INJ": {"1": -0.25, "2": 0.0}, "PROD": {"2": -0.5}}

tests/testdata/well_trajectory/read_laterals/mlt_p2_a.json

@@ -0,0 +1 @@
+{"INJ": {"1": 0.5, "2": 0.5}, "PROD": {"2": 0.5}}

tests/testdata/well_trajectory/read_laterals/mlt_p2_b.json

@@ -0,0 +1 @@
+{"INJ": {"1": 0.0, "2": 0.0}, "PROD": {"2": 0.0}}

tests/testdata/well_trajectory/read_laterals/mlt_p2_c.json

@@ -0,0 +1 @@
+{"INJ": {"1": 0.5, "2": 0.5}, "PROD": {"2": 0.5}}

tests/testdata/well_trajectory/read_laterals/mlt_p3_x.json

@@ -0,0 +1 @@
+{"INJ": {"1": -0.45, "2": -0.25}, "PROD": {"2": 0.5}}

tests/testdata/well_trajectory/read_laterals/mlt_p3_y.json

@@ -0,0 +1 @@
+{"INJ": {"1": -0.25, "2": 0.45}, "PROD": {"2": 0.5}}

tests/testdata/well_trajectory/read_laterals/mlt_p3_z.json

@@ -0,0 +1 @@
+{"INJ": {"1": -0.25, "2": 0.0}, "PROD": {"2": 0.5}}

tests/testdata/well_trajectory/read_laterals/p1_x.json

@@ -0,0 +1 @@
+{"INJ": -0.25, "PROD": 0.25}

tests/testdata/well_trajectory/read_laterals/p1_y.json

@@ -0,0 +1 @@
+{"INJ": -0.25, "PROD": 0.25}

tests/testdata/well_trajectory/read_laterals/p1_z.json

@@ -0,0 +1 @@
+{"INJ": -0.5, "PROD": -0.5}

tests/testdata/well_trajectory/read_laterals/p2_a.json

@@ -0,0 +1 @@
+{"INJ": 0.5, "PROD": 0.5}

tests/testdata/well_trajectory/read_laterals/p2_b.json

@@ -0,0 +1 @@
+{"INJ": 0.0, "PROD": 0.0}

tests/testdata/well_trajectory/read_laterals/p2_c.json

@@ -0,0 +1 @@
+{"INJ": 0.5, "PROD": 0.5}

tests/testdata/well_trajectory/read_laterals/p3_x.json

@@ -0,0 +1 @@
+{"INJ": -0.25001, "PROD": 0.25001}

tests/testdata/well_trajectory/read_laterals/p3_y.json

@@ -0,0 +1 @@
+{"INJ": -0.25, "PROD": 0.25}

tests/testdata/well_trajectory/read_laterals/p3_z.json

@@ -0,0 +1 @@
+{"INJ": 0.5, "PROD": 0.5}