Add optimize float folds test

expr-lang · Nov 8, 2022 · 4fc6235 · 4fc6235
1 parent b26d4b7
commit 4fc6235
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/optimizer/optimizer_test.go b/optimizer/optimizer_test.go
@@ -28,6 +28,18 @@ func TestOptimize_constant_folding(t *testing.T) {
 	assert.Equal(t, ast.Dump(expected), ast.Dump(tree.Node))
 }
 
+func TestOptimize_constant_folding_with_floats(t *testing.T) {
+	tree, err := parser.Parse(`1 + 2.0 * ((1.0 * 2) / 2) - 0`)
+	require.NoError(t, err)
+
+	err = optimizer.Optimize(&tree.Node, nil)
+	require.NoError(t, err)
+
+	expected := &ast.FloatNode{Value: 3.0}
+
+	assert.Equal(t, ast.Dump(expected), ast.Dump(tree.Node))
+}
+
 func TestOptimize_in_array(t *testing.T) {
 	config := conf.New(map[string]int{"v": 0})