From 7e7488af2eb1cbbaf04df49bc4a5e42912f093e0 Mon Sep 17 00:00:00 2001 From: Bel LaPointe Date: Fri, 27 Oct 2023 19:28:48 -0600 Subject: [PATCH] export existing contribution predictor --- ana/predictor.go | 4 ++-- ana/predictor_test.go | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/ana/predictor.go b/ana/predictor.go index 001a469..d5c941b 100644 --- a/ana/predictor.go +++ b/ana/predictor.go @@ -44,10 +44,10 @@ func NewInterestPredictor(namePattern string, currencyPattern string, apy float6 func NewAutoContributionPredictor(reg ledger.Register) Predictor { monthlyRate := getMonthlyAutoContributionRates(reg) - return newAutoContributionPredictor(monthlyRate) + return NewContributionPredictor(monthlyRate) } -func newAutoContributionPredictor(monthlyRate map[string]ledger.Balance) Predictor { +func NewContributionPredictor(monthlyRate ledger.Balances) Predictor { return func(given ledger.Balances, delta time.Duration) ledger.Balances { months := float64(delta) / float64(month) result := make(ledger.Balances) diff --git a/ana/predictor_test.go b/ana/predictor_test.go index 0d88ffb..ae51964 100644 --- a/ana/predictor_test.go +++ b/ana/predictor_test.go @@ -98,10 +98,10 @@ func TestGetMonthlyAutoContributionRate(t *testing.T) { } } -func TestNewAutoContributionPredictor(t *testing.T) { +func TestNewContributionPredictor(t *testing.T) { name := "x" currency := ledger.USD - predictor := newAutoContributionPredictor(map[string]ledger.Balance{ + predictor := NewContributionPredictor(map[string]ledger.Balance{ name: {currency: 10}, "y": {"XYZ": 3}, })