230 lines
6.8 KiB
Go
230 lines
6.8 KiB
Go
// Copyright 2020 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package strconv_test
|
|
|
|
import (
|
|
"math"
|
|
"math/cmplx"
|
|
"reflect"
|
|
. "strconv"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
infp0 = complex(math.Inf(+1), 0)
|
|
infm0 = complex(math.Inf(-1), 0)
|
|
inf0p = complex(0, math.Inf(+1))
|
|
inf0m = complex(0, math.Inf(-1))
|
|
|
|
infpp = complex(math.Inf(+1), math.Inf(+1))
|
|
infpm = complex(math.Inf(+1), math.Inf(-1))
|
|
infmp = complex(math.Inf(-1), math.Inf(+1))
|
|
infmm = complex(math.Inf(-1), math.Inf(-1))
|
|
)
|
|
|
|
type atocTest struct {
|
|
in string
|
|
out complex128
|
|
err error
|
|
}
|
|
|
|
func TestParseComplex(t *testing.T) {
|
|
tests := []atocTest{
|
|
// Clearly invalid
|
|
{"", 0, ErrSyntax},
|
|
{" ", 0, ErrSyntax},
|
|
{"(", 0, ErrSyntax},
|
|
{")", 0, ErrSyntax},
|
|
{"i", 0, ErrSyntax},
|
|
{"+i", 0, ErrSyntax},
|
|
{"-i", 0, ErrSyntax},
|
|
{"1I", 0, ErrSyntax},
|
|
{"10 + 5i", 0, ErrSyntax},
|
|
{"3+", 0, ErrSyntax},
|
|
{"3+5", 0, ErrSyntax},
|
|
{"3+5+5i", 0, ErrSyntax},
|
|
|
|
// Parentheses
|
|
{"()", 0, ErrSyntax},
|
|
{"(i)", 0, ErrSyntax},
|
|
{"(0)", 0, nil},
|
|
{"(1i)", 1i, nil},
|
|
{"(3.0+5.5i)", 3.0 + 5.5i, nil},
|
|
{"(1)+1i", 0, ErrSyntax},
|
|
{"(3.0+5.5i", 0, ErrSyntax},
|
|
{"3.0+5.5i)", 0, ErrSyntax},
|
|
|
|
// NaNs
|
|
{"NaN", complex(math.NaN(), 0), nil},
|
|
{"NANi", complex(0, math.NaN()), nil},
|
|
{"nan+nAni", complex(math.NaN(), math.NaN()), nil},
|
|
{"+NaN", 0, ErrSyntax},
|
|
{"-NaN", 0, ErrSyntax},
|
|
{"NaN-NaNi", 0, ErrSyntax},
|
|
|
|
// Infs
|
|
{"Inf", infp0, nil},
|
|
{"+inf", infp0, nil},
|
|
{"-inf", infm0, nil},
|
|
{"Infinity", infp0, nil},
|
|
{"+INFINITY", infp0, nil},
|
|
{"-infinity", infm0, nil},
|
|
{"+infi", inf0p, nil},
|
|
{"0-infinityi", inf0m, nil},
|
|
{"Inf+Infi", infpp, nil},
|
|
{"+Inf-Infi", infpm, nil},
|
|
{"-Infinity+Infi", infmp, nil},
|
|
{"inf-inf", 0, ErrSyntax},
|
|
|
|
// Zeros
|
|
{"0", 0, nil},
|
|
{"0i", 0, nil},
|
|
{"-0.0i", 0, nil},
|
|
{"0+0.0i", 0, nil},
|
|
{"0e+0i", 0, nil},
|
|
{"0e-0+0i", 0, nil},
|
|
{"-0.0-0.0i", 0, nil},
|
|
{"0e+012345", 0, nil},
|
|
{"0x0p+012345i", 0, nil},
|
|
{"0x0.00p-012345i", 0, nil},
|
|
{"+0e-0+0e-0i", 0, nil},
|
|
{"0e+0+0e+0i", 0, nil},
|
|
{"-0e+0-0e+0i", 0, nil},
|
|
|
|
// Regular non-zeroes
|
|
{"0.1", 0.1, nil},
|
|
{"0.1i", 0 + 0.1i, nil},
|
|
{"0.123", 0.123, nil},
|
|
{"0.123i", 0 + 0.123i, nil},
|
|
{"0.123+0.123i", 0.123 + 0.123i, nil},
|
|
{"99", 99, nil},
|
|
{"+99", 99, nil},
|
|
{"-99", -99, nil},
|
|
{"+1i", 1i, nil},
|
|
{"-1i", -1i, nil},
|
|
{"+3+1i", 3 + 1i, nil},
|
|
{"30+3i", 30 + 3i, nil},
|
|
{"+3e+3-3e+3i", 3e+3 - 3e+3i, nil},
|
|
{"+3e+3+3e+3i", 3e+3 + 3e+3i, nil},
|
|
{"+3e+3+3e+3i+", 0, ErrSyntax},
|
|
|
|
// Separators
|
|
{"0.1", 0.1, nil},
|
|
{"0.1i", 0 + 0.1i, nil},
|
|
{"0.1_2_3", 0.123, nil},
|
|
{"+0x_3p3i", 0x3p3i, nil},
|
|
{"0_0+0x_0p0i", 0, nil},
|
|
{"0x_10.3p-8+0x3p3i", 0x10.3p-8 + 0x3p3i, nil},
|
|
{"+0x_1_0.3p-8+0x_3_0p3i", 0x10.3p-8 + 0x30p3i, nil},
|
|
{"0x1_0.3p+8-0x_3p3i", 0x10.3p+8 - 0x3p3i, nil},
|
|
|
|
// Hexadecimals
|
|
{"0x10.3p-8+0x3p3i", 0x10.3p-8 + 0x3p3i, nil},
|
|
{"+0x10.3p-8+0x3p3i", 0x10.3p-8 + 0x3p3i, nil},
|
|
{"0x10.3p+8-0x3p3i", 0x10.3p+8 - 0x3p3i, nil},
|
|
{"0x1p0", 1, nil},
|
|
{"0x1p1", 2, nil},
|
|
{"0x1p-1", 0.5, nil},
|
|
{"0x1ep-1", 15, nil},
|
|
{"-0x1ep-1", -15, nil},
|
|
{"-0x2p3", -16, nil},
|
|
{"0x1e2", 0, ErrSyntax},
|
|
{"1p2", 0, ErrSyntax},
|
|
{"0x1e2i", 0, ErrSyntax},
|
|
|
|
// ErrRange
|
|
// next float64 - too large
|
|
{"+0x1p1024", infp0, ErrRange},
|
|
{"-0x1p1024", infm0, ErrRange},
|
|
{"+0x1p1024i", inf0p, ErrRange},
|
|
{"-0x1p1024i", inf0m, ErrRange},
|
|
{"+0x1p1024+0x1p1024i", infpp, ErrRange},
|
|
{"+0x1p1024-0x1p1024i", infpm, ErrRange},
|
|
{"-0x1p1024+0x1p1024i", infmp, ErrRange},
|
|
{"-0x1p1024-0x1p1024i", infmm, ErrRange},
|
|
// the border is ...158079
|
|
// borderline - okay
|
|
{"+0x1.fffffffffffff7fffp1023+0x1.fffffffffffff7fffp1023i", 1.7976931348623157e+308 + 1.7976931348623157e+308i, nil},
|
|
{"+0x1.fffffffffffff7fffp1023-0x1.fffffffffffff7fffp1023i", 1.7976931348623157e+308 - 1.7976931348623157e+308i, nil},
|
|
{"-0x1.fffffffffffff7fffp1023+0x1.fffffffffffff7fffp1023i", -1.7976931348623157e+308 + 1.7976931348623157e+308i, nil},
|
|
{"-0x1.fffffffffffff7fffp1023-0x1.fffffffffffff7fffp1023i", -1.7976931348623157e+308 - 1.7976931348623157e+308i, nil},
|
|
// borderline - too large
|
|
{"+0x1.fffffffffffff8p1023", infp0, ErrRange},
|
|
{"-0x1fffffffffffff.8p+971", infm0, ErrRange},
|
|
{"+0x1.fffffffffffff8p1023i", inf0p, ErrRange},
|
|
{"-0x1fffffffffffff.8p+971i", inf0m, ErrRange},
|
|
{"+0x1.fffffffffffff8p1023+0x1.fffffffffffff8p1023i", infpp, ErrRange},
|
|
{"+0x1.fffffffffffff8p1023-0x1.fffffffffffff8p1023i", infpm, ErrRange},
|
|
{"-0x1fffffffffffff.8p+971+0x1fffffffffffff.8p+971i", infmp, ErrRange},
|
|
{"-0x1fffffffffffff8p+967-0x1fffffffffffff8p+967i", infmm, ErrRange},
|
|
// a little too large
|
|
{"1e308+1e308i", 1e+308 + 1e+308i, nil},
|
|
{"2e308+2e308i", infpp, ErrRange},
|
|
{"1e309+1e309i", infpp, ErrRange},
|
|
{"0x1p1025+0x1p1025i", infpp, ErrRange},
|
|
{"2e308", infp0, ErrRange},
|
|
{"1e309", infp0, ErrRange},
|
|
{"0x1p1025", infp0, ErrRange},
|
|
{"2e308i", inf0p, ErrRange},
|
|
{"1e309i", inf0p, ErrRange},
|
|
{"0x1p1025i", inf0p, ErrRange},
|
|
// way too large
|
|
{"+1e310+1e310i", infpp, ErrRange},
|
|
{"+1e310-1e310i", infpm, ErrRange},
|
|
{"-1e310+1e310i", infmp, ErrRange},
|
|
{"-1e310-1e310i", infmm, ErrRange},
|
|
// under/overflow exponent
|
|
{"1e-4294967296", 0, nil},
|
|
{"1e-4294967296i", 0, nil},
|
|
{"1e-4294967296+1i", 1i, nil},
|
|
{"1+1e-4294967296i", 1, nil},
|
|
{"1e-4294967296+1e-4294967296i", 0, nil},
|
|
{"1e+4294967296", infp0, ErrRange},
|
|
{"1e+4294967296i", inf0p, ErrRange},
|
|
{"1e+4294967296+1e+4294967296i", infpp, ErrRange},
|
|
{"1e+4294967296-1e+4294967296i", infpm, ErrRange},
|
|
}
|
|
for i := range tests {
|
|
test := &tests[i]
|
|
if test.err != nil {
|
|
test.err = &NumError{Func: "ParseComplex", Num: test.in, Err: test.err}
|
|
}
|
|
got, err := ParseComplex(test.in, 128)
|
|
if !reflect.DeepEqual(err, test.err) {
|
|
t.Fatalf("ParseComplex(%q, 128) = %v, %v; want %v, %v", test.in, got, err, test.out, test.err)
|
|
}
|
|
if !(cmplx.IsNaN(test.out) && cmplx.IsNaN(got)) && got != test.out {
|
|
t.Fatalf("ParseComplex(%q, 128) = %v, %v; want %v, %v", test.in, got, err, test.out, test.err)
|
|
}
|
|
|
|
if complex128(complex64(test.out)) == test.out {
|
|
got, err := ParseComplex(test.in, 64)
|
|
if !reflect.DeepEqual(err, test.err) {
|
|
t.Fatalf("ParseComplex(%q, 64) = %v, %v; want %v, %v", test.in, got, err, test.out, test.err)
|
|
}
|
|
got64 := complex64(got)
|
|
if complex128(got64) != test.out {
|
|
t.Fatalf("ParseComplex(%q, 64) = %v, %v; want %v, %v", test.in, got, err, test.out, test.err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Issue 42297: allow ParseComplex(s, not_32_or_64) for legacy reasons
|
|
func TestParseComplexIncorrectBitSize(t *testing.T) {
|
|
const s = "1.5e308+1.0e307i"
|
|
const want = 1.5e308 + 1.0e307i
|
|
|
|
for _, bitSize := range []int{0, 10, 100, 256} {
|
|
c, err := ParseComplex(s, bitSize)
|
|
if err != nil {
|
|
t.Fatalf("ParseComplex(%q, %d) gave error %s", s, bitSize, err)
|
|
}
|
|
if c != want {
|
|
t.Fatalf("ParseComplex(%q, %d) = %g (expected %g)", s, bitSize, c, want)
|
|
}
|
|
}
|
|
}
|