Add test cases where target is polyline in Fault.connect

equinor · Sep 3, 2024 · 010fedc · 010fedc
1 parent e2b8348
commit 010fedc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/tests/rd_tests/test_faults.py b/python/tests/rd_tests/test_faults.py
@@ -775,9 +775,13 @@ def test_connectWithPolyline(self):
         fault3.addRecord(1, 1, 0, 2, 0, 0, "X")
 
         self.assertIsNone(fault3.connect(fault1, 0))
+        self.assertIsNone(fault3.connect(fault1.getPolyline(0), 0))
+        self.assertIsNone(fault3.connect(Polyline(init_points=[(4, 4), (1, 2)]), 0))
 
         intersect = fault2.connect(fault1, 0)
+        intersect2 = fault2.connect(Polyline(init_points=[(0, 0), (1, 2)]), 0)
         self.assertEqual(len(intersect), 2)
+        self.assertEqual(len(intersect2), 2)
         p1 = intersect[0]
         p2 = intersect[1]