35 lines
557 B
Go
35 lines
557 B
Go
package core
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
)
|
|
|
|
func TestGetStatsForTZs(t *testing.T) {
|
|
preds, err := GetStatsForTZs([]string{"+0100", "+0000"})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(preds) == 0 {
|
|
t.Fatalf("failed to get predictions")
|
|
}
|
|
}
|
|
|
|
func TestPredSum(t *testing.T) {
|
|
preds, err := GetStatsForTZs([]string{"+0100"})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// sum must be 1
|
|
sum := 0.0
|
|
for _, pred := range preds {
|
|
sum += pred.Probability
|
|
}
|
|
|
|
if math.Abs(sum-1) > 1e-6 {
|
|
t.Fatalf("Probabilities did not sum to 1. Got %.4f instead", sum)
|
|
}
|
|
}
|