adding GetLastBoundedValue and tests
This commit is contained in:
parent
d888440415
commit
f8573f1123
2 changed files with 47 additions and 0 deletions
|
@ -81,6 +81,34 @@ func (bbs *BollingerBandsSeries) GetBoundedValue(index int) (x, y1, y2 float64)
|
|||
return
|
||||
}
|
||||
|
||||
// GetLastBoundedValue returns the last bounded value for the series.
|
||||
func (bbs *BollingerBandsSeries) GetLastBoundedValue() (x, y1, y2 float64) {
|
||||
if bbs.InnerSeries == nil {
|
||||
return
|
||||
}
|
||||
windowSize := bbs.GetWindowSize()
|
||||
seriesLength := bbs.InnerSeries.Len()
|
||||
startAt := seriesLength - windowSize
|
||||
if startAt < 0 {
|
||||
startAt = 0
|
||||
}
|
||||
|
||||
vb := NewRingBufferWithCapacity(windowSize)
|
||||
for index := startAt; index < seriesLength; index++ {
|
||||
xn, yn := bbs.InnerSeries.GetValue(index)
|
||||
vb.Enqueue(yn)
|
||||
x = xn
|
||||
}
|
||||
|
||||
ay := bbs.getAverage(vb)
|
||||
std := bbs.getStdDev(vb)
|
||||
|
||||
y1 = ay + (bbs.GetK() * std)
|
||||
y2 = ay - (bbs.GetK() * std)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// Render renders the series.
|
||||
func (bbs BollingerBandsSeries) Render(r Renderer, canvasBox Box, xrange, yrange Range, defaults Style) {
|
||||
s := bbs.Style.WithDefaultsFrom(defaults)
|
||||
|
|
|
@ -1,6 +1,7 @@
|
|||
package chart
|
||||
|
||||
import (
|
||||
"math"
|
||||
"testing"
|
||||
|
||||
"github.com/blendlabs/go-assert"
|
||||
|
@ -30,3 +31,21 @@ func TestBollingerBandSeries(t *testing.T) {
|
|||
assert.True(y1values[x] > y2values[x])
|
||||
}
|
||||
}
|
||||
|
||||
func TestBollingerBandLastValue(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
s1 := mockValueProvider{
|
||||
X: Seq(1.0, 100.0),
|
||||
Y: Seq(1.0, 100.0),
|
||||
}
|
||||
|
||||
bbs := &BollingerBandsSeries{
|
||||
InnerSeries: s1,
|
||||
}
|
||||
|
||||
x, y1, y2 := bbs.GetLastBoundedValue()
|
||||
assert.Equal(100.0, x)
|
||||
assert.Equal(100, math.Floor(y1))
|
||||
assert.Equal(95, math.Floor(y2))
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue