1 files changed, 24 insertions, 0 deletions
diff --git a/common/pretty/bytesize_test.go b/common/pretty/bytesize_test.go
new file mode 100644
index 0000000..b83f9c0
--- /dev/null
+++ b/common/pretty/bytesize_test.go
@@ -0,0 +1,24 @@
+package pretty
+
+import (
+	"testing"
+)
+
+func TestBytes(t *testing.T) {
+	tests := []struct {
+		Expected string
+		Actual   string
+	}{
+		{"128 Byte", Bytes(128)},
+		{"128.00 KiB", Bytes(128 * 1024)},
+		{"128.00 MiB", Bytes(128 * 1024 * 1024)},
+		{"128.00 GiB", Bytes(128 * 1024 * 1024 * 1024)},
+		{"128.00 TiB", Bytes(128 * 1024 * 1024 * 1024 * 1024)},
+		{"131072.00 TiB", Bytes(128 * 1024 * 1024 * 1024 * 1024 * 1024)},
+	}
+	for i, tc := range tests {
+		if tc.Expected != tc.Actual {
+			t.Fatalf("Test %v failed: %v", i, tc)
+		}
+	}
+}