123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129 |
- package test
- import (
- "bytes"
- "encoding/json"
- "fmt"
- "github.com/json-iterator/go"
- "github.com/stretchr/testify/require"
- "strconv"
- "testing"
- )
- func Test_read_float(t *testing.T) {
- inputs := []string{
- `1.1`, `1000`, `9223372036854775807`, `12.3`, `-12.3`, `720368.54775807`, `720368.547758075`,
- `1e1`, `1e+1`, `1e-1`, `1E1`, `1E+1`, `1E-1`, `-1e1`, `-1e+1`, `-1e-1`,
- }
- for _, input := range inputs {
- // non-streaming
- t.Run(fmt.Sprintf("%v", input), func(t *testing.T) {
- should := require.New(t)
- iter := jsoniter.ParseString(jsoniter.ConfigDefault, input+",")
- expected, err := strconv.ParseFloat(input, 32)
- should.Nil(err)
- should.Equal(float32(expected), iter.ReadFloat32())
- })
- t.Run(fmt.Sprintf("%v", input), func(t *testing.T) {
- should := require.New(t)
- iter := jsoniter.ParseString(jsoniter.ConfigDefault, input+",")
- expected, err := strconv.ParseFloat(input, 64)
- should.Nil(err)
- should.Equal(expected, iter.ReadFloat64())
- })
- // streaming
- t.Run(fmt.Sprintf("%v", input), func(t *testing.T) {
- should := require.New(t)
- iter := jsoniter.Parse(jsoniter.ConfigDefault, bytes.NewBufferString(input+","), 2)
- expected, err := strconv.ParseFloat(input, 32)
- should.Nil(err)
- should.Equal(float32(expected), iter.ReadFloat32())
- })
- t.Run(fmt.Sprintf("%v", input), func(t *testing.T) {
- should := require.New(t)
- iter := jsoniter.Parse(jsoniter.ConfigDefault, bytes.NewBufferString(input+","), 2)
- val := float64(0)
- err := json.Unmarshal([]byte(input), &val)
- should.Nil(err)
- should.Equal(val, iter.ReadFloat64())
- })
- }
- }
- func Test_write_float32(t *testing.T) {
- vals := []float32{0, 1, -1, 99, 0xff, 0xfff, 0xffff, 0xfffff, 0xffffff, 0x4ffffff, 0xfffffff,
- -0x4ffffff, -0xfffffff, 1.2345, 1.23456, 1.234567, 1.001}
- for _, val := range vals {
- t.Run(fmt.Sprintf("%v", val), func(t *testing.T) {
- should := require.New(t)
- buf := &bytes.Buffer{}
- stream := jsoniter.NewStream(jsoniter.ConfigDefault, buf, 4096)
- stream.WriteFloat32Lossy(val)
- stream.Flush()
- should.Nil(stream.Error)
- output, err := json.Marshal(val)
- should.Nil(err)
- should.Equal(string(output), buf.String())
- })
- t.Run(fmt.Sprintf("%v", val), func(t *testing.T) {
- should := require.New(t)
- buf := &bytes.Buffer{}
- stream := jsoniter.NewStream(jsoniter.ConfigDefault, buf, 4096)
- stream.WriteVal(val)
- stream.Flush()
- should.Nil(stream.Error)
- output, err := json.Marshal(val)
- should.Nil(err)
- should.Equal(string(output), buf.String())
- })
- }
- should := require.New(t)
- buf := &bytes.Buffer{}
- stream := jsoniter.NewStream(jsoniter.ConfigDefault, buf, 10)
- stream.WriteRaw("abcdefg")
- stream.WriteFloat32Lossy(1.123456)
- stream.Flush()
- should.Nil(stream.Error)
- should.Equal("abcdefg1.123456", buf.String())
- stream = jsoniter.NewStream(jsoniter.ConfigDefault, nil, 0)
- stream.WriteFloat32(float32(0.0000001))
- should.Equal("1e-07", string(stream.Buffer()))
- }
- func Test_write_float64(t *testing.T) {
- vals := []float64{0, 1, -1, 99, 0xff, 0xfff, 0xffff, 0xfffff, 0xffffff, 0x4ffffff, 0xfffffff,
- -0x4ffffff, -0xfffffff, 1.2345, 1.23456, 1.234567, 1.001}
- for _, val := range vals {
- t.Run(fmt.Sprintf("%v", val), func(t *testing.T) {
- should := require.New(t)
- buf := &bytes.Buffer{}
- stream := jsoniter.NewStream(jsoniter.ConfigDefault, buf, 4096)
- stream.WriteFloat64Lossy(val)
- stream.Flush()
- should.Nil(stream.Error)
- should.Equal(strconv.FormatFloat(val, 'f', -1, 64), buf.String())
- })
- t.Run(fmt.Sprintf("%v", val), func(t *testing.T) {
- should := require.New(t)
- buf := &bytes.Buffer{}
- stream := jsoniter.NewStream(jsoniter.ConfigDefault, buf, 4096)
- stream.WriteVal(val)
- stream.Flush()
- should.Nil(stream.Error)
- should.Equal(strconv.FormatFloat(val, 'f', -1, 64), buf.String())
- })
- }
- should := require.New(t)
- buf := &bytes.Buffer{}
- stream := jsoniter.NewStream(jsoniter.ConfigDefault, buf, 10)
- stream.WriteRaw("abcdefg")
- stream.WriteFloat64Lossy(1.123456)
- stream.Flush()
- should.Nil(stream.Error)
- should.Equal("abcdefg1.123456", buf.String())
- stream = jsoniter.NewStream(jsoniter.ConfigDefault, nil, 0)
- stream.WriteFloat64(float64(0.0000001))
- should.Equal("1e-07", string(stream.Buffer()))
- }
|