xgo/objects_test.go
daniel a9a233a750
fix internal package issue (#241)
* fix internal package issue

* ExampleSimple -> Example
2019-12-24 07:42:30 -08:00

736 lines
21 KiB
Go

package tengo_test
import (
"testing"
"github.com/d5/tengo"
"github.com/d5/tengo/require"
"github.com/d5/tengo/token"
)
func TestObject_TypeName(t *testing.T) {
var o tengo.Object = &tengo.Int{}
require.Equal(t, "int", o.TypeName())
o = &tengo.Float{}
require.Equal(t, "float", o.TypeName())
o = &tengo.Char{}
require.Equal(t, "char", o.TypeName())
o = &tengo.String{}
require.Equal(t, "string", o.TypeName())
o = &tengo.Bool{}
require.Equal(t, "bool", o.TypeName())
o = &tengo.Array{}
require.Equal(t, "array", o.TypeName())
o = &tengo.Map{}
require.Equal(t, "map", o.TypeName())
o = &tengo.ArrayIterator{}
require.Equal(t, "array-iterator", o.TypeName())
o = &tengo.StringIterator{}
require.Equal(t, "string-iterator", o.TypeName())
o = &tengo.MapIterator{}
require.Equal(t, "map-iterator", o.TypeName())
o = &tengo.BuiltinFunction{Name: "fn"}
require.Equal(t, "builtin-function:fn", o.TypeName())
o = &tengo.UserFunction{Name: "fn"}
require.Equal(t, "user-function:fn", o.TypeName())
o = &tengo.CompiledFunction{}
require.Equal(t, "compiled-function", o.TypeName())
o = &tengo.Undefined{}
require.Equal(t, "undefined", o.TypeName())
o = &tengo.Error{}
require.Equal(t, "error", o.TypeName())
o = &tengo.Bytes{}
require.Equal(t, "bytes", o.TypeName())
}
func TestObject_IsFalsy(t *testing.T) {
var o tengo.Object = &tengo.Int{Value: 0}
require.True(t, o.IsFalsy())
o = &tengo.Int{Value: 1}
require.False(t, o.IsFalsy())
o = &tengo.Float{Value: 0}
require.False(t, o.IsFalsy())
o = &tengo.Float{Value: 1}
require.False(t, o.IsFalsy())
o = &tengo.Char{Value: ' '}
require.False(t, o.IsFalsy())
o = &tengo.Char{Value: 'T'}
require.False(t, o.IsFalsy())
o = &tengo.String{Value: ""}
require.True(t, o.IsFalsy())
o = &tengo.String{Value: " "}
require.False(t, o.IsFalsy())
o = &tengo.Array{Value: nil}
require.True(t, o.IsFalsy())
o = &tengo.Array{Value: []tengo.Object{nil}} // nil is not valid but still count as 1 element
require.False(t, o.IsFalsy())
o = &tengo.Map{Value: nil}
require.True(t, o.IsFalsy())
o = &tengo.Map{Value: map[string]tengo.Object{"a": nil}} // nil is not valid but still count as 1 element
require.False(t, o.IsFalsy())
o = &tengo.StringIterator{}
require.True(t, o.IsFalsy())
o = &tengo.ArrayIterator{}
require.True(t, o.IsFalsy())
o = &tengo.MapIterator{}
require.True(t, o.IsFalsy())
o = &tengo.BuiltinFunction{}
require.False(t, o.IsFalsy())
o = &tengo.CompiledFunction{}
require.False(t, o.IsFalsy())
o = &tengo.Undefined{}
require.True(t, o.IsFalsy())
o = &tengo.Error{}
require.True(t, o.IsFalsy())
o = &tengo.Bytes{}
require.True(t, o.IsFalsy())
o = &tengo.Bytes{Value: []byte{1, 2}}
require.False(t, o.IsFalsy())
}
func TestObject_String(t *testing.T) {
var o tengo.Object = &tengo.Int{Value: 0}
require.Equal(t, "0", o.String())
o = &tengo.Int{Value: 1}
require.Equal(t, "1", o.String())
o = &tengo.Float{Value: 0}
require.Equal(t, "0", o.String())
o = &tengo.Float{Value: 1}
require.Equal(t, "1", o.String())
o = &tengo.Char{Value: ' '}
require.Equal(t, " ", o.String())
o = &tengo.Char{Value: 'T'}
require.Equal(t, "T", o.String())
o = &tengo.String{Value: ""}
require.Equal(t, `""`, o.String())
o = &tengo.String{Value: " "}
require.Equal(t, `" "`, o.String())
o = &tengo.Array{Value: nil}
require.Equal(t, "[]", o.String())
o = &tengo.Map{Value: nil}
require.Equal(t, "{}", o.String())
o = &tengo.Error{Value: nil}
require.Equal(t, "error", o.String())
o = &tengo.Error{Value: &tengo.String{Value: "error 1"}}
require.Equal(t, `error: "error 1"`, o.String())
o = &tengo.StringIterator{}
require.Equal(t, "<string-iterator>", o.String())
o = &tengo.ArrayIterator{}
require.Equal(t, "<array-iterator>", o.String())
o = &tengo.MapIterator{}
require.Equal(t, "<map-iterator>", o.String())
o = &tengo.Undefined{}
require.Equal(t, "<undefined>", o.String())
o = &tengo.Bytes{}
require.Equal(t, "", o.String())
o = &tengo.Bytes{Value: []byte("foo")}
require.Equal(t, "foo", o.String())
}
func TestObject_BinaryOp(t *testing.T) {
var o tengo.Object = &tengo.Char{}
_, err := o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
o = &tengo.Bool{}
_, err = o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
o = &tengo.Map{}
_, err = o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
o = &tengo.ArrayIterator{}
_, err = o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
o = &tengo.StringIterator{}
_, err = o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
o = &tengo.MapIterator{}
_, err = o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
o = &tengo.BuiltinFunction{}
_, err = o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
o = &tengo.CompiledFunction{}
_, err = o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
o = &tengo.Undefined{}
_, err = o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
o = &tengo.Error{}
_, err = o.BinaryOp(token.Add, tengo.UndefinedValue)
require.Error(t, err)
}
func TestArray_BinaryOp(t *testing.T) {
testBinaryOp(t, &tengo.Array{Value: nil}, token.Add,
&tengo.Array{Value: nil}, &tengo.Array{Value: nil})
testBinaryOp(t, &tengo.Array{Value: nil}, token.Add,
&tengo.Array{Value: []tengo.Object{}}, &tengo.Array{Value: nil})
testBinaryOp(t, &tengo.Array{Value: []tengo.Object{}}, token.Add,
&tengo.Array{Value: nil}, &tengo.Array{Value: []tengo.Object{}})
testBinaryOp(t, &tengo.Array{Value: []tengo.Object{}}, token.Add,
&tengo.Array{Value: []tengo.Object{}},
&tengo.Array{Value: []tengo.Object{}})
testBinaryOp(t, &tengo.Array{Value: nil}, token.Add,
&tengo.Array{Value: []tengo.Object{
&tengo.Int{Value: 1},
}}, &tengo.Array{Value: []tengo.Object{
&tengo.Int{Value: 1},
}})
testBinaryOp(t, &tengo.Array{Value: nil}, token.Add,
&tengo.Array{Value: []tengo.Object{
&tengo.Int{Value: 1},
&tengo.Int{Value: 2},
&tengo.Int{Value: 3},
}}, &tengo.Array{Value: []tengo.Object{
&tengo.Int{Value: 1},
&tengo.Int{Value: 2},
&tengo.Int{Value: 3},
}})
testBinaryOp(t, &tengo.Array{Value: []tengo.Object{
&tengo.Int{Value: 1},
&tengo.Int{Value: 2},
&tengo.Int{Value: 3},
}}, token.Add, &tengo.Array{Value: nil},
&tengo.Array{Value: []tengo.Object{
&tengo.Int{Value: 1},
&tengo.Int{Value: 2},
&tengo.Int{Value: 3},
}})
testBinaryOp(t, &tengo.Array{Value: []tengo.Object{
&tengo.Int{Value: 1},
&tengo.Int{Value: 2},
&tengo.Int{Value: 3},
}}, token.Add, &tengo.Array{Value: []tengo.Object{
&tengo.Int{Value: 4},
&tengo.Int{Value: 5},
&tengo.Int{Value: 6},
}}, &tengo.Array{Value: []tengo.Object{
&tengo.Int{Value: 1},
&tengo.Int{Value: 2},
&tengo.Int{Value: 3},
&tengo.Int{Value: 4},
&tengo.Int{Value: 5},
&tengo.Int{Value: 6},
}})
}
func TestError_Equals(t *testing.T) {
err1 := &tengo.Error{Value: &tengo.String{Value: "some error"}}
err2 := err1
require.True(t, err1.Equals(err2))
require.True(t, err2.Equals(err1))
err2 = &tengo.Error{Value: &tengo.String{Value: "some error"}}
require.False(t, err1.Equals(err2))
require.False(t, err2.Equals(err1))
}
func TestFloat_BinaryOp(t *testing.T) {
// float + float
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := float64(-2); r <= 2.1; r += 0.4 {
testBinaryOp(t, &tengo.Float{Value: l}, token.Add,
&tengo.Float{Value: r}, &tengo.Float{Value: l + r})
}
}
// float - float
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := float64(-2); r <= 2.1; r += 0.4 {
testBinaryOp(t, &tengo.Float{Value: l}, token.Sub,
&tengo.Float{Value: r}, &tengo.Float{Value: l - r})
}
}
// float * float
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := float64(-2); r <= 2.1; r += 0.4 {
testBinaryOp(t, &tengo.Float{Value: l}, token.Mul,
&tengo.Float{Value: r}, &tengo.Float{Value: l * r})
}
}
// float / float
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := float64(-2); r <= 2.1; r += 0.4 {
if r != 0 {
testBinaryOp(t, &tengo.Float{Value: l}, token.Quo,
&tengo.Float{Value: r}, &tengo.Float{Value: l / r})
}
}
}
// float < float
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := float64(-2); r <= 2.1; r += 0.4 {
testBinaryOp(t, &tengo.Float{Value: l}, token.Less,
&tengo.Float{Value: r}, boolValue(l < r))
}
}
// float > float
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := float64(-2); r <= 2.1; r += 0.4 {
testBinaryOp(t, &tengo.Float{Value: l}, token.Greater,
&tengo.Float{Value: r}, boolValue(l > r))
}
}
// float <= float
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := float64(-2); r <= 2.1; r += 0.4 {
testBinaryOp(t, &tengo.Float{Value: l}, token.LessEq,
&tengo.Float{Value: r}, boolValue(l <= r))
}
}
// float >= float
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := float64(-2); r <= 2.1; r += 0.4 {
testBinaryOp(t, &tengo.Float{Value: l}, token.GreaterEq,
&tengo.Float{Value: r}, boolValue(l >= r))
}
}
// float + int
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Float{Value: l}, token.Add,
&tengo.Int{Value: r}, &tengo.Float{Value: l + float64(r)})
}
}
// float - int
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Float{Value: l}, token.Sub,
&tengo.Int{Value: r}, &tengo.Float{Value: l - float64(r)})
}
}
// float * int
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Float{Value: l}, token.Mul,
&tengo.Int{Value: r}, &tengo.Float{Value: l * float64(r)})
}
}
// float / int
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := int64(-2); r <= 2; r++ {
if r != 0 {
testBinaryOp(t, &tengo.Float{Value: l}, token.Quo,
&tengo.Int{Value: r},
&tengo.Float{Value: l / float64(r)})
}
}
}
// float < int
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Float{Value: l}, token.Less,
&tengo.Int{Value: r}, boolValue(l < float64(r)))
}
}
// float > int
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Float{Value: l}, token.Greater,
&tengo.Int{Value: r}, boolValue(l > float64(r)))
}
}
// float <= int
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Float{Value: l}, token.LessEq,
&tengo.Int{Value: r}, boolValue(l <= float64(r)))
}
}
// float >= int
for l := float64(-2); l <= 2.1; l += 0.4 {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Float{Value: l}, token.GreaterEq,
&tengo.Int{Value: r}, boolValue(l >= float64(r)))
}
}
}
func TestInt_BinaryOp(t *testing.T) {
// int + int
for l := int64(-2); l <= 2; l++ {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Int{Value: l}, token.Add,
&tengo.Int{Value: r}, &tengo.Int{Value: l + r})
}
}
// int - int
for l := int64(-2); l <= 2; l++ {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Int{Value: l}, token.Sub,
&tengo.Int{Value: r}, &tengo.Int{Value: l - r})
}
}
// int * int
for l := int64(-2); l <= 2; l++ {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Int{Value: l}, token.Mul,
&tengo.Int{Value: r}, &tengo.Int{Value: l * r})
}
}
// int / int
for l := int64(-2); l <= 2; l++ {
for r := int64(-2); r <= 2; r++ {
if r != 0 {
testBinaryOp(t, &tengo.Int{Value: l}, token.Quo,
&tengo.Int{Value: r}, &tengo.Int{Value: l / r})
}
}
}
// int % int
for l := int64(-4); l <= 4; l++ {
for r := -int64(-4); r <= 4; r++ {
if r == 0 {
testBinaryOp(t, &tengo.Int{Value: l}, token.Rem,
&tengo.Int{Value: r}, &tengo.Int{Value: l % r})
}
}
}
// int & int
testBinaryOp(t,
&tengo.Int{Value: 0}, token.And, &tengo.Int{Value: 0},
&tengo.Int{Value: int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.And, &tengo.Int{Value: 0},
&tengo.Int{Value: int64(1) & int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 0}, token.And, &tengo.Int{Value: 1},
&tengo.Int{Value: int64(0) & int64(1)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.And, &tengo.Int{Value: 1},
&tengo.Int{Value: int64(1)})
testBinaryOp(t,
&tengo.Int{Value: 0}, token.And, &tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(0) & int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.And, &tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(1) & int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: int64(0xffffffff)}, token.And,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: 1984}, token.And,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(1984) & int64(0xffffffff)})
testBinaryOp(t, &tengo.Int{Value: -1984}, token.And,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(-1984) & int64(0xffffffff)})
// int | int
testBinaryOp(t,
&tengo.Int{Value: 0}, token.Or, &tengo.Int{Value: 0},
&tengo.Int{Value: int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.Or, &tengo.Int{Value: 0},
&tengo.Int{Value: int64(1) | int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 0}, token.Or, &tengo.Int{Value: 1},
&tengo.Int{Value: int64(0) | int64(1)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.Or, &tengo.Int{Value: 1},
&tengo.Int{Value: int64(1)})
testBinaryOp(t,
&tengo.Int{Value: 0}, token.Or, &tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(0) | int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.Or, &tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(1) | int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: int64(0xffffffff)}, token.Or,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: 1984}, token.Or,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(1984) | int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: -1984}, token.Or,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(-1984) | int64(0xffffffff)})
// int ^ int
testBinaryOp(t,
&tengo.Int{Value: 0}, token.Xor, &tengo.Int{Value: 0},
&tengo.Int{Value: int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.Xor, &tengo.Int{Value: 0},
&tengo.Int{Value: int64(1) ^ int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 0}, token.Xor, &tengo.Int{Value: 1},
&tengo.Int{Value: int64(0) ^ int64(1)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.Xor, &tengo.Int{Value: 1},
&tengo.Int{Value: int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 0}, token.Xor, &tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(0) ^ int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.Xor, &tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(1) ^ int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: int64(0xffffffff)}, token.Xor,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 1984}, token.Xor,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(1984) ^ int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: -1984}, token.Xor,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(-1984) ^ int64(0xffffffff)})
// int &^ int
testBinaryOp(t,
&tengo.Int{Value: 0}, token.AndNot, &tengo.Int{Value: 0},
&tengo.Int{Value: int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.AndNot, &tengo.Int{Value: 0},
&tengo.Int{Value: int64(1) &^ int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 0}, token.AndNot,
&tengo.Int{Value: 1}, &tengo.Int{Value: int64(0) &^ int64(1)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.AndNot, &tengo.Int{Value: 1},
&tengo.Int{Value: int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 0}, token.AndNot,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(0) &^ int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.AndNot,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(1) &^ int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: int64(0xffffffff)}, token.AndNot,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(0)})
testBinaryOp(t,
&tengo.Int{Value: 1984}, token.AndNot,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(1984) &^ int64(0xffffffff)})
testBinaryOp(t,
&tengo.Int{Value: -1984}, token.AndNot,
&tengo.Int{Value: int64(0xffffffff)},
&tengo.Int{Value: int64(-1984) &^ int64(0xffffffff)})
// int << int
for s := int64(0); s < 64; s++ {
testBinaryOp(t,
&tengo.Int{Value: 0}, token.Shl, &tengo.Int{Value: s},
&tengo.Int{Value: int64(0) << uint(s)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.Shl, &tengo.Int{Value: s},
&tengo.Int{Value: int64(1) << uint(s)})
testBinaryOp(t,
&tengo.Int{Value: 2}, token.Shl, &tengo.Int{Value: s},
&tengo.Int{Value: int64(2) << uint(s)})
testBinaryOp(t,
&tengo.Int{Value: -1}, token.Shl, &tengo.Int{Value: s},
&tengo.Int{Value: int64(-1) << uint(s)})
testBinaryOp(t,
&tengo.Int{Value: -2}, token.Shl, &tengo.Int{Value: s},
&tengo.Int{Value: int64(-2) << uint(s)})
testBinaryOp(t,
&tengo.Int{Value: int64(0xffffffff)}, token.Shl,
&tengo.Int{Value: s},
&tengo.Int{Value: int64(0xffffffff) << uint(s)})
}
// int >> int
for s := int64(0); s < 64; s++ {
testBinaryOp(t,
&tengo.Int{Value: 0}, token.Shr, &tengo.Int{Value: s},
&tengo.Int{Value: int64(0) >> uint(s)})
testBinaryOp(t,
&tengo.Int{Value: 1}, token.Shr, &tengo.Int{Value: s},
&tengo.Int{Value: int64(1) >> uint(s)})
testBinaryOp(t,
&tengo.Int{Value: 2}, token.Shr, &tengo.Int{Value: s},
&tengo.Int{Value: int64(2) >> uint(s)})
testBinaryOp(t,
&tengo.Int{Value: -1}, token.Shr, &tengo.Int{Value: s},
&tengo.Int{Value: int64(-1) >> uint(s)})
testBinaryOp(t,
&tengo.Int{Value: -2}, token.Shr, &tengo.Int{Value: s},
&tengo.Int{Value: int64(-2) >> uint(s)})
testBinaryOp(t,
&tengo.Int{Value: int64(0xffffffff)}, token.Shr,
&tengo.Int{Value: s},
&tengo.Int{Value: int64(0xffffffff) >> uint(s)})
}
// int < int
for l := int64(-2); l <= 2; l++ {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Int{Value: l}, token.Less,
&tengo.Int{Value: r}, boolValue(l < r))
}
}
// int > int
for l := int64(-2); l <= 2; l++ {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Int{Value: l}, token.Greater,
&tengo.Int{Value: r}, boolValue(l > r))
}
}
// int <= int
for l := int64(-2); l <= 2; l++ {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Int{Value: l}, token.LessEq,
&tengo.Int{Value: r}, boolValue(l <= r))
}
}
// int >= int
for l := int64(-2); l <= 2; l++ {
for r := int64(-2); r <= 2; r++ {
testBinaryOp(t, &tengo.Int{Value: l}, token.GreaterEq,
&tengo.Int{Value: r}, boolValue(l >= r))
}
}
// int + float
for l := int64(-2); l <= 2; l++ {
for r := float64(-2); r <= 2.1; r += 0.5 {
testBinaryOp(t, &tengo.Int{Value: l}, token.Add,
&tengo.Float{Value: r},
&tengo.Float{Value: float64(l) + r})
}
}
// int - float
for l := int64(-2); l <= 2; l++ {
for r := float64(-2); r <= 2.1; r += 0.5 {
testBinaryOp(t, &tengo.Int{Value: l}, token.Sub,
&tengo.Float{Value: r},
&tengo.Float{Value: float64(l) - r})
}
}
// int * float
for l := int64(-2); l <= 2; l++ {
for r := float64(-2); r <= 2.1; r += 0.5 {
testBinaryOp(t, &tengo.Int{Value: l}, token.Mul,
&tengo.Float{Value: r},
&tengo.Float{Value: float64(l) * r})
}
}
// int / float
for l := int64(-2); l <= 2; l++ {
for r := float64(-2); r <= 2.1; r += 0.5 {
if r != 0 {
testBinaryOp(t, &tengo.Int{Value: l}, token.Quo,
&tengo.Float{Value: r},
&tengo.Float{Value: float64(l) / r})
}
}
}
// int < float
for l := int64(-2); l <= 2; l++ {
for r := float64(-2); r <= 2.1; r += 0.5 {
testBinaryOp(t, &tengo.Int{Value: l}, token.Less,
&tengo.Float{Value: r}, boolValue(float64(l) < r))
}
}
// int > float
for l := int64(-2); l <= 2; l++ {
for r := float64(-2); r <= 2.1; r += 0.5 {
testBinaryOp(t, &tengo.Int{Value: l}, token.Greater,
&tengo.Float{Value: r}, boolValue(float64(l) > r))
}
}
// int <= float
for l := int64(-2); l <= 2; l++ {
for r := float64(-2); r <= 2.1; r += 0.5 {
testBinaryOp(t, &tengo.Int{Value: l}, token.LessEq,
&tengo.Float{Value: r}, boolValue(float64(l) <= r))
}
}
// int >= float
for l := int64(-2); l <= 2; l++ {
for r := float64(-2); r <= 2.1; r += 0.5 {
testBinaryOp(t, &tengo.Int{Value: l}, token.GreaterEq,
&tengo.Float{Value: r}, boolValue(float64(l) >= r))
}
}
}
func TestMap_Index(t *testing.T) {
m := &tengo.Map{Value: make(map[string]tengo.Object)}
k := &tengo.Int{Value: 1}
v := &tengo.String{Value: "abcdef"}
err := m.IndexSet(k, v)
require.NoError(t, err)
res, err := m.IndexGet(k)
require.NoError(t, err)
require.Equal(t, v, res)
}
func TestString_BinaryOp(t *testing.T) {
lstr := "abcde"
rstr := "01234"
for l := 0; l < len(lstr); l++ {
for r := 0; r < len(rstr); r++ {
ls := lstr[l:]
rs := rstr[r:]
testBinaryOp(t, &tengo.String{Value: ls}, token.Add,
&tengo.String{Value: rs},
&tengo.String{Value: ls + rs})
rc := []rune(rstr)[r]
testBinaryOp(t, &tengo.String{Value: ls}, token.Add,
&tengo.Char{Value: rc},
&tengo.String{Value: ls + string(rc)})
}
}
}
func testBinaryOp(
t *testing.T,
lhs tengo.Object,
op token.Token,
rhs tengo.Object,
expected tengo.Object,
) {
t.Helper()
actual, err := lhs.BinaryOp(op, rhs)
require.NoError(t, err)
require.Equal(t, expected, actual)
}
func boolValue(b bool) tengo.Object {
if b {
return tengo.TrueValue
}
return tengo.FalseValue
}