58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package ana
|
|
|
|
import (
|
|
"fmt"
|
|
"maps"
|
|
"testing"
|
|
"time"
|
|
|
|
"gogs.inhome.blapointe.com/ana-ledger/ledger"
|
|
)
|
|
|
|
func TestPredictionPredict(t *testing.T) {
|
|
inc := func(b ledger.Balances, _ time.Duration) ledger.Balances {
|
|
result := make(ledger.Balances)
|
|
for k, v := range b {
|
|
result[k] = maps.Clone(v)
|
|
for k2, v2 := range result[k] {
|
|
result[k][k2] = v2 + 1
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
double := func(b ledger.Balances, _ time.Duration) ledger.Balances {
|
|
result := make(ledger.Balances)
|
|
for k, v := range b {
|
|
result[k] = maps.Clone(v)
|
|
for k2, v2 := range result[k] {
|
|
result[k][k2] = v2 * 2
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
from := time.Now()
|
|
these := []time.Time{time.Now().Add(time.Hour * 24 * 365 / 12)}
|
|
theseK := these[0].Format("2006-01")
|
|
_, _ = inc, double
|
|
|
|
cases := map[string]struct {
|
|
prediction Prediction
|
|
given ledger.Balances
|
|
want ledger.Register
|
|
}{
|
|
"empty": {
|
|
want: ledger.Register{theseK: {}},
|
|
},
|
|
}
|
|
|
|
for name, d := range cases {
|
|
c := d
|
|
t.Run(name, func(t *testing.T) {
|
|
got := c.prediction.predict(c.given, from, these)
|
|
if fmt.Sprintf("%+v", got) != fmt.Sprintf("%+v", c.want) {
|
|
t.Errorf("want\n\t%+v, got\n\t%+v", c.want, got)
|
|
}
|
|
})
|
|
}
|
|
}
|