114 lines
2 KiB
Go
114 lines
2 KiB
Go
|
package varfmt
|
||
|
|
||
|
import (
|
||
|
"strings"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/stretchr/testify/require"
|
||
|
)
|
||
|
|
||
|
func TestFormat(t *testing.T) {
|
||
|
for _, tc := range []struct {
|
||
|
format string
|
||
|
args []interface{}
|
||
|
result string
|
||
|
}{
|
||
|
{
|
||
|
format: "{}",
|
||
|
args: []interface{}{0},
|
||
|
result: "0",
|
||
|
},
|
||
|
{
|
||
|
format: "{0} {0}",
|
||
|
args: []interface{}{1},
|
||
|
result: "1 1",
|
||
|
},
|
||
|
{
|
||
|
format: "{1} {5}",
|
||
|
args: []interface{}{0, 1, 2, 3, 4, 5, 6},
|
||
|
result: "1 5",
|
||
|
},
|
||
|
{
|
||
|
format: "{} {} {} {} {}",
|
||
|
args: []interface{}{0, 1, 2, 3, 4},
|
||
|
result: "0 1 2 3 4",
|
||
|
},
|
||
|
{
|
||
|
format: "{} {0} {0} {0} {}",
|
||
|
args: []interface{}{0, 1, 2, 3, 4},
|
||
|
result: "0 0 0 0 4",
|
||
|
},
|
||
|
{
|
||
|
format: "Hello, {2}",
|
||
|
args: []interface{}{0, 1, "World"},
|
||
|
result: "Hello, World",
|
||
|
},
|
||
|
} {
|
||
|
t.Run(tc.result, func(t *testing.T) {
|
||
|
require.Equal(t, tc.result, Sprintf(tc.format, tc.args...))
|
||
|
})
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func BenchmarkFormat(b *testing.B) {
|
||
|
for _, tc := range []struct {
|
||
|
name string
|
||
|
format string
|
||
|
args []interface{}
|
||
|
}{
|
||
|
{
|
||
|
name: "small int",
|
||
|
format: "{}",
|
||
|
args: []interface{}{42},
|
||
|
},
|
||
|
{
|
||
|
name: "small string",
|
||
|
format: "{} {}",
|
||
|
args: []interface{}{"Hello", "World"},
|
||
|
},
|
||
|
{
|
||
|
name: "big",
|
||
|
format: strings.Repeat("{0}{1}", 1000),
|
||
|
args: []interface{}{42, 43},
|
||
|
},
|
||
|
} {
|
||
|
b.Run(tc.name, func(b *testing.B) {
|
||
|
b.ReportAllocs()
|
||
|
for i := 0; i < b.N; i++ {
|
||
|
_ = Sprintf(tc.format, tc.args...)
|
||
|
}
|
||
|
})
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func BenchmarkSprintf(b *testing.B) {
|
||
|
for _, tc := range []struct {
|
||
|
name string
|
||
|
format string
|
||
|
args []interface{}
|
||
|
}{
|
||
|
{
|
||
|
name: "small",
|
||
|
format: "%d",
|
||
|
args: []interface{}{42},
|
||
|
},
|
||
|
{
|
||
|
name: "small string",
|
||
|
format: "%v %v",
|
||
|
args: []interface{}{"Hello", "World"},
|
||
|
}, {
|
||
|
name: "big",
|
||
|
format: strings.Repeat("%[0]v%[1]v", 1000),
|
||
|
args: []interface{}{42, 43},
|
||
|
},
|
||
|
} {
|
||
|
b.Run(tc.name, func(b *testing.B) {
|
||
|
b.ReportAllocs()
|
||
|
|
||
|
for i := 0; i < b.N; i++ {
|
||
|
_ = Sprintf(tc.format, tc.args...)
|
||
|
}
|
||
|
})
|
||
|
}
|
||
|
}
|