py: handle corner cases for str.find

sbinet · sbinet · commit 4b114ff8e742 · 2022-05-24T19:23:34.000+02:00
Signed-off-by: Sebastien Binet &lt;binet@cern.ch&gt;
diff --git a/py/string.go b/py/string.go
@@ -602,11 +602,18 @@ func (s String) find(args Tuple) (Object, error) {
 	}
 
 	var (
-		beg = int(pybeg.(Int))
-		end = int(pyend.(Int))
+		beg  = int(pybeg.(Int))
+		end  = int(pyend.(Int))
+		size = s.len()
 	)
+	if beg > size {
+		beg = size
+	}
 	if end < 0 {
-		end = s.len()
+		end = size
+	}
+	if end > size {
+		end = size
 	}
 
 	var (
diff --git a/py/string_test.go b/py/string_test.go
@@ -67,6 +67,20 @@ func TestStringFind(t *testing.T) {
 			end: -1,
 			idx: 6,
 		},
+		{
+			str: "0123456789",
+			sub: "6",
+			beg: 100,
+			end: -1,
+			idx: -1,
+		},
+		{
+			str: "0123456789",
+			sub: "6",
+			beg: 2,
+			end: 1,
+			idx: -1,
+		},
 	} {
 		t.Run(tc.str+":"+tc.sub, func(t *testing.T) {
 			beg := tc.beg
