Fix floating points with underscores
diff --git a/lexer_test.go b/lexer_test.go
index 82946bf..d8abb35 100644
--- a/lexer_test.go
+++ b/lexer_test.go
@@ -464,6 +464,13 @@
token{Position{1, 7}, tokenInteger, "1_2_3_4_5"},
token{Position{1, 16}, tokenEOF, ""},
})
+
+ testFlow(t, "flt8 = 9_224_617.445_991_228_313", []token{
+ token{Position{1, 1}, tokenKey, "flt8"},
+ token{Position{1, 6}, tokenEqual, "="},
+ token{Position{1, 8}, tokenFloat, "9_224_617.445_991_228_313"},
+ token{Position{1, 33}, tokenEOF, ""},
+ })
}
func TestMultiline(t *testing.T) {
diff --git a/parser.go b/parser.go
index 59f513b..7ba872b 100644
--- a/parser.go
+++ b/parser.go
@@ -231,7 +231,8 @@
}
return val
case tokenFloat:
- val, err := strconv.ParseFloat(tok.val, 64)
+ cleanedVal := strings.Replace(tok.val, "_", "", -1)
+ val, err := strconv.ParseFloat(cleanedVal, 64)
if err != nil {
p.raiseError(tok, "%s", err)
}
diff --git a/parser_test.go b/parser_test.go
index ebf1626..94308a4 100644
--- a/parser_test.go
+++ b/parser_test.go
@@ -68,7 +68,7 @@
})
}
-func TestIntegersWithUnderscores(t *testing.T) {
+func TestNumbersWithUnderscores(t *testing.T) {
tree, err := Load("a = 1_000")
assertTree(t, tree, err, map[string]interface{}{
"a": int64(1000),
@@ -83,6 +83,16 @@
assertTree(t, tree, err, map[string]interface{}{
"a": int64(12345),
})
+
+ tree, err = Load("flt8 = 9_224_617.445_991_228_313")
+ assertTree(t, tree, err, map[string]interface{}{
+ "flt8": float64(9224617.445991228313),
+ })
+
+ tree, err = Load("flt9 = 1e1_00")
+ assertTree(t, tree, err, map[string]interface{}{
+ "flt9": float64(1e100),
+ })
}
func TestFloatsWithExponents(t *testing.T) {
