Convert math params to number instead of failing with error

yuin · Feb 3, 2023 · 410e9bd · 410e9bd
1 parent fa815b5
commit 410e9bd
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+.idea
diff --git a/auxlib.go b/auxlib.go
@@ -40,6 +40,11 @@ func (ls *LState) CheckNumber(n int) LNumber {
 	if lv, ok := v.(LNumber); ok {
 		return lv
 	}
+	if lv, ok := v.(LString); ok {
+		if num, err := parseNumber(string(lv)); err == nil {
+			return num
+		}
+	}
 	ls.TypeError(n, LTNumber)
 	return 0
 }

diff --git a/auxlib_test.go b/auxlib_test.go
@@ -36,8 +36,10 @@ func TestCheckNumber(t *testing.T) {
 	errorIfGFuncNotFail(t, L, func(L *LState) int {
 		L.Push(LNumber(10))
 		errorIfNotEqual(t, LNumber(10), L.CheckNumber(2))
+		L.Push(LString("11"))
+		errorIfNotEqual(t, LNumber(11), L.CheckNumber(3))
 		L.Push(LString("aaa"))
-		L.CheckNumber(3)
+		L.CheckNumber(4)
 		return 0
 	}, "number expected, got string")
 }