py: handle end=-1 in str.find

sbinet · sbinet · commit e4b6e829e68e · 2022-05-24T19:19:58.000+02:00
Signed-off-by: Sebastien Binet &lt;binet@cern.ch&gt;
diff --git a/py/string.go b/py/string.go
@@ -593,7 +593,7 @@ func (s String) find(args Tuple) (Object, error) {
 	var (
 		pysub Object
 		pybeg Object = Int(0)
-		pyend Object = Int(len(s))
+		pyend Object = Int(s.len())
 		pyfmt        = "s|ii:find"
 	)
 	err := ParseTuple(args, pyfmt, &pysub, &pybeg, &pyend)
@@ -604,6 +604,12 @@ func (s String) find(args Tuple) (Object, error) {
 	var (
 		beg = int(pybeg.(Int))
 		end = int(pyend.(Int))
+	)
+	if end < 0 {
+		end = s.len()
+	}
+
+	var (
 		off = s.slice(0, beg, s.len()).len()
 		str = string(s.slice(beg, end, s.len()))
 		sub = string(pysub.(String))
diff --git a/py/string_test.go b/py/string_test.go
@@ -60,6 +60,13 @@ func TestStringFind(t *testing.T) {
 			end: 7,
 			idx: 6,
 		},
+		{
+			str: "0123456789",
+			sub: "6",
+			beg: 1,
+			end: -1,
+			idx: 6,
+		},
 	} {
 		t.Run(tc.str+":"+tc.sub, func(t *testing.T) {
 			beg := tc.beg
