File: units_test.go

package info (click to toggle)
golang-github-jedib0t-go-pretty 6.5.9-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,700 kB
  • sloc: makefile: 28; sh: 14
file content (46 lines) | stat: -rw-r--r-- 1,660 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
package progress

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestFormatBytes(t *testing.T) {
	assert.Equal(t, "1B", FormatBytes(1))
	assert.Equal(t, "1.50KB", FormatBytes(1500))
	assert.Equal(t, "1.50MB", FormatBytes(1500000))
	assert.Equal(t, "1.50GB", FormatBytes(1500000000))
	assert.Equal(t, "1.50TB", FormatBytes(1500000000000))
	assert.Equal(t, "1.50PB", FormatBytes(1500000000000000))
	assert.Equal(t, "1500.00PB", FormatBytes(1500000000000000000))
}

func TestFormatNumber(t *testing.T) {
	assert.Equal(t, "1", FormatNumber(1))
	assert.Equal(t, "1.50K", FormatNumber(1500))
	assert.Equal(t, "1.50M", FormatNumber(1500000))
	assert.Equal(t, "1.50B", FormatNumber(1500000000))
	assert.Equal(t, "1.50T", FormatNumber(1500000000000))
	assert.Equal(t, "1.50Q", FormatNumber(1500000000000000))
	assert.Equal(t, "1500.00Q", FormatNumber(1500000000000000000))
}

func TestUnits_Sprint(t *testing.T) {
	assert.Equal(t, "1.50K", UnitsDefault.Sprint(1500))
	assert.Equal(t, "1.50KB", UnitsBytes.Sprint(1500))
	assert.Equal(t, "$1.50K", UnitsCurrencyDollar.Sprint(1500))
	assert.Equal(t, "₠1.50K", UnitsCurrencyEuro.Sprint(1500))
	assert.Equal(t, "£1.50K", UnitsCurrencyPound.Sprint(1500))

	customUnits := Units{Notation: "#"}
	assert.Equal(t, "#1.50K", customUnits.Sprint(1500))
}

func TestUnits_NotationPosition(t *testing.T) {
	afterUnits := Units{Notation: " ₽", NotationPosition: UnitsNotationPositionAfter}
	assert.Equal(t, "1.50K ₽", afterUnits.Sprint(1500))

	unknownNotationPosition := Units{Notation: "* ", NotationPosition: UnitsNotationPosition(999)}
	assert.Equal(t, "* 1.50K", unknownNotationPosition.Sprint(1500))
}