mirror of https://github.com/ollama/ollama.git
37 lines
633 B
Go
37 lines
633 B
Go
package format
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestHumanNumber(t *testing.T) {
|
|
type testCase struct {
|
|
input uint64
|
|
expected string
|
|
}
|
|
|
|
testCases := []testCase{
|
|
{0, "0"},
|
|
{999, "999"},
|
|
{1000, "1K"},
|
|
{1001, "1K"},
|
|
{1000000, "1M"},
|
|
{125000000, "125M"},
|
|
{500500000, "500.50M"},
|
|
{500550000, "500.55M"},
|
|
{1000000000, "1B"},
|
|
{2800000000, "2.8B"},
|
|
{2850000000, "2.9B"},
|
|
{1000000000000, "1000B"},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.expected, func(t *testing.T) {
|
|
result := HumanNumber(tc.input)
|
|
if result != tc.expected {
|
|
t.Errorf("Expected %s, got %s", tc.expected, result)
|
|
}
|
|
})
|
|
}
|
|
}
|