go-chart/exp_moving_average_series_test.go
2016-07-16 13:25:21 -07:00

36 lines
640 B
Go

package chart
import (
"math"
"testing"
"github.com/blendlabs/go-assert"
)
func TestExponentialMovingAverageSeries(t *testing.T) {
assert := assert.New(t)
mockSeries := mockValueProvider{
Seq(1.0, 10.0),
Seq(10, 1.0),
}
assert.Equal(10, mockSeries.Len())
mas := &ExponentialMovingAverageSeries{
InnerSeries: mockSeries,
}
var yvalues []float64
for x := 0; x < mas.Len(); x++ {
_, y := mas.GetValue(x)
yvalues = append(yvalues, y)
}
assert.Equal(10.0, yvalues[0])
assert.True(math.Abs(yvalues[9]-3.77) < 0.01)
lvx, lvy := mas.GetLastValue()
assert.Equal(10.0, lvx)
assert.True(math.Abs(lvy-3.77) < 0.01)
}