bel/ana-ledger
1
0
Fork 0
Code Actions Packages Projects Releases Activity

i am so pleased

Browse Source
This commit is contained in:
bel
2023-10-26 22:20:48 -06:00
parent 54940fab52
commit 8d004c44ce
2 changed files with 49 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
ana/predictor_test.go
View File

@@ -105,3 +105,33 @@ func TestGetMonthlyContributionRate(t *testing.T) {
t.Error(got["y"])
}
}
func TestNewContributionPredictor(t *testing.T) {
name := "x"
currency := ledger.USD
predictor := newContributionPredictor(map[string]ledger.Balance{
name: {currency: 10},
"y": {"XYZ": 3},
})
month := time.Hour * 24 * 365 / 12
if got := predictor(ledger.Balances{}, 2*month); got[name][currency] != 20 {
t.Error(got[name])
} else if got["y"]["XYZ"] != 6 {
t.Error(got["y"])
}
if got := predictor(ledger.Balances{name: {currency: 30}}, 2*month); got[name][currency] != 30+20 {
t.Error(got)
} else if got["y"]["XYZ"] != 6 {
t.Error(got["y"])
}
if got := predictor(ledger.Balances{"z": {"ABC": 100}}, 2*month); got[name][currency] != 20 {
t.Error(got)
} else if got["y"]["XYZ"] != 6 {
t.Error(got["y"])
} else if got["z"]["ABC"] != 100 {
t.Error(got["z"])
}
}