util_test.go

package xgp

import (
	"fmt"
	"testing"
	"time"
)

func TestCountDistinct(t *testing.T) {
	var (
		testCases = []struct {
			x []float64
			n int
		}{
			{
				x: []float64{1, 2, 3},
				n: 3,
			},
			{
				x: []float64{1, 1, 3},
				n: 2,
			},
			{
				x: []float64{1, 1, 1},
				n: 1,
			},
			{
				x: []float64{},
				n: 0,
			},
		}
	)
	for i, tc := range testCases {
		t.Run(fmt.Sprintf("TC %d", i), func(t *testing.T) {
			var n = countDistinct(tc.x)
			if n != tc.n {
				t.Errorf("Expected %d, got %d", tc.n, n)
			}
		})
	}
}

func TestFmtDuration(t *testing.T) {
	var (
		testCases = []struct {
			d time.Duration
			s string
		}{
			{
				d: time.Duration(42) * time.Second,
				s: "00:00:42",
			},
			{
				d: time.Duration(42+60*42) * time.Second,
				s: "00:42:42",
			},
			{
				d: time.Duration(42+60*42+42*60*60) * time.Second,
				s: "42:42:42",
			},
			{
				d: time.Duration(42) * time.Nanosecond,
				s: "00:00:00",
			},
			{
				d: time.Duration(42+60*42+42*60*60*60) * time.Second,
				s: "2520:42:42",
			},
		}
	)
	for i, tc := range testCases {
		t.Run(fmt.Sprintf("TC %d", i), func(t *testing.T) {
			var s = fmtDuration(tc.d)
			if s != tc.s {
				t.Errorf("Expected %s, got %s", tc.s, s)
			}
		})
	}
}