mirror of
https://github.com/go-sylixos/elvish.git
synced 2024-12-05 03:17:50 +08:00
111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
package vals
|
|
|
|
import (
|
|
"math"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"src.elv.sh/pkg/testutil"
|
|
. "src.elv.sh/pkg/tt"
|
|
)
|
|
|
|
// Test utilities.
|
|
|
|
const (
|
|
zeros = "0000000000000000000"
|
|
// Values that exceed the range of int64, used for testing BigInt.
|
|
z = "1" + zeros + "0"
|
|
z1 = "1" + zeros + "1" // z+1
|
|
z2 = "1" + zeros + "2" // z+2
|
|
z3 = "1" + zeros + "3" // z+3
|
|
zz = "2" + zeros + "0" // 2z
|
|
zz1 = "2" + zeros + "1" // 2z+1
|
|
zz2 = "2" + zeros + "2" // 2z+2
|
|
zz3 = "2" + zeros + "3" // 2z+3
|
|
)
|
|
|
|
func TestParseNum(t *testing.T) {
|
|
Test(t, Fn("ParseNum", ParseNum), Table{
|
|
Args("1").Rets(1),
|
|
|
|
Args(z).Rets(bigInt(z)),
|
|
|
|
Args("1/2").Rets(big.NewRat(1, 2)),
|
|
Args("2/1").Rets(2),
|
|
Args(z + "/1").Rets(bigInt(z)),
|
|
|
|
Args("1.0").Rets(1.0),
|
|
Args("1e-5").Rets(1e-5),
|
|
|
|
Args("x").Rets(nil),
|
|
Args("x/y").Rets(nil),
|
|
})
|
|
}
|
|
|
|
func TestUnifyNums(t *testing.T) {
|
|
Test(t, Fn("UnifyNums", UnifyNums), Table{
|
|
Args([]Num{1, 2, 3, 4}, Int).
|
|
Rets([]int{1, 2, 3, 4}),
|
|
|
|
Args([]Num{1, 2, 3, bigInt(z)}, Int).
|
|
Rets([]*big.Int{big.NewInt(1), big.NewInt(2), big.NewInt(3), bigInt(z)}),
|
|
|
|
Args([]Num{1, 2, 3, big.NewRat(1, 2)}, Int).
|
|
Rets([]*big.Rat{
|
|
big.NewRat(1, 1), big.NewRat(2, 1),
|
|
big.NewRat(3, 1), big.NewRat(1, 2)}),
|
|
Args([]Num{1, 2, bigInt(z), big.NewRat(1, 2)}, Int).
|
|
Rets([]*big.Rat{
|
|
big.NewRat(1, 1), big.NewRat(2, 1), bigRat(z), big.NewRat(1, 2)}),
|
|
|
|
Args([]Num{1, 2, 3, 4.0}, Int).
|
|
Rets([]float64{1, 2, 3, 4}),
|
|
Args([]Num{1, 2, big.NewRat(1, 2), 4.0}, Int).
|
|
Rets([]float64{1, 2, 0.5, 4}),
|
|
Args([]Num{1, 2, big.NewInt(3), 4.0}, Int).
|
|
Rets([]float64{1, 2, 3, 4}),
|
|
Args([]Num{1, 2, bigInt(z), 4.0}, Int).
|
|
Rets([]float64{1, 2, math.Inf(1), 4}),
|
|
|
|
Args([]Num{1, 2, 3, 4}, BigInt).
|
|
Rets([]*big.Int{
|
|
big.NewInt(1), big.NewInt(2), big.NewInt(3), big.NewInt(4)}),
|
|
})
|
|
}
|
|
|
|
func TestUnifyNums2(t *testing.T) {
|
|
Test(t, Fn("UnifyNums2", UnifyNums2), Table{
|
|
Args(1, 2, Int).Rets(1, 2),
|
|
Args(1, bigInt(z), Int).Rets(big.NewInt(1), bigInt(z)),
|
|
Args(1, big.NewRat(1, 2), Int).Rets(big.NewRat(1, 1), big.NewRat(1, 2)),
|
|
Args(1, 2.0, Int).Rets(1.0, 2.0),
|
|
|
|
Args(1, 2, BigInt).Rets(big.NewInt(1), big.NewInt(2)),
|
|
})
|
|
}
|
|
|
|
func TestInvalidNumType(t *testing.T) {
|
|
Test(t, Fn("Recover", testutil.Recover), Table{
|
|
Args(func() { UnifyNums([]Num{int32(0)}, 0) }).Rets("invalid num type int32"),
|
|
Args(func() { PromoteToBigInt(int32(0)) }).Rets("invalid num type int32"),
|
|
Args(func() { PromoteToBigRat(int32(0)) }).Rets("invalid num type int32"),
|
|
Args(func() { ConvertToFloat64(int32(0)) }).Rets("invalid num type int32"),
|
|
})
|
|
}
|
|
|
|
func bigInt(s string) *big.Int {
|
|
z, ok := new(big.Int).SetString(s, 0)
|
|
if !ok {
|
|
panic("cannot parse as big int: " + s)
|
|
}
|
|
return z
|
|
}
|
|
|
|
func bigRat(s string) *big.Rat {
|
|
z, ok := new(big.Rat).SetString(s)
|
|
if !ok {
|
|
panic("cannot parse as big rat: " + s)
|
|
}
|
|
return z
|
|
}
|