@@ -405,6 +405,9 @@ func checkAndRemoveUnderscoresFloats(b []byte) ([]byte, error) {
|
|||||||
if !before {
|
if !before {
|
||||||
return nil, newDecodeError(b[i-1:i+1], "number must have at least one digit between underscores")
|
return nil, newDecodeError(b[i-1:i+1], "number must have at least one digit between underscores")
|
||||||
}
|
}
|
||||||
|
if i < len(b)-1 && (b[i+1] == 'e' || b[i+1] == 'E') {
|
||||||
|
return nil, newDecodeError(b[i+1:i+2], "cannot have underscore before exponent")
|
||||||
|
}
|
||||||
before = false
|
before = false
|
||||||
case 'e', 'E':
|
case 'e', 'E':
|
||||||
if i < len(b)-1 && b[i+1] == '_' {
|
if i < len(b)-1 && b[i+1] == '_' {
|
||||||
|
|||||||
@@ -204,6 +204,16 @@ func TestUnmarshal_Floats(t *testing.T) {
|
|||||||
assert.True(t, math.IsNaN(v))
|
assert.True(t, math.IsNaN(v))
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
desc: "underscore after integer part",
|
||||||
|
input: `1_e2`,
|
||||||
|
err: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
desc: "underscore after integer part",
|
||||||
|
input: `1.0_e2`,
|
||||||
|
err: true,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
type doc struct {
|
type doc struct {
|
||||||
@@ -215,12 +225,16 @@ func TestUnmarshal_Floats(t *testing.T) {
|
|||||||
t.Run(e.desc, func(t *testing.T) {
|
t.Run(e.desc, func(t *testing.T) {
|
||||||
doc := doc{}
|
doc := doc{}
|
||||||
err := toml.Unmarshal([]byte(`A = `+e.input), &doc)
|
err := toml.Unmarshal([]byte(`A = `+e.input), &doc)
|
||||||
|
if e.err {
|
||||||
|
require.Error(t, err)
|
||||||
|
} else {
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
if e.testFn != nil {
|
if e.testFn != nil {
|
||||||
e.testFn(t, doc.A)
|
e.testFn(t, doc.A)
|
||||||
} else {
|
} else {
|
||||||
assert.Equal(t, e.expected, doc.A)
|
assert.Equal(t, e.expected, doc.A)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user