111 lines
2.4 KiB
Go
111 lines
2.4 KiB
Go
package chart
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/wcharczuk/go-chart/v2/testutil"
|
|
)
|
|
|
|
type mockValuesProvider struct {
|
|
X []float64
|
|
Y []float64
|
|
}
|
|
|
|
func (m mockValuesProvider) Len() int {
|
|
return MinInt(len(m.X), len(m.Y))
|
|
}
|
|
|
|
func (m mockValuesProvider) GetValues(index int) (x, y float64) {
|
|
if index < 0 {
|
|
panic("negative index at GetValue()")
|
|
}
|
|
if index >= MinInt(len(m.X), len(m.Y)) {
|
|
panic("index is outside the length of m.X or m.Y")
|
|
}
|
|
x = m.X[index]
|
|
y = m.Y[index]
|
|
return
|
|
}
|
|
|
|
func TestSMASeriesGetValue(t *testing.T) {
|
|
// replaced new assertions helper
|
|
|
|
mockSeries := mockValuesProvider{
|
|
LinearRange(1.0, 10.0),
|
|
LinearRange(10, 1.0),
|
|
}
|
|
testutil.AssertEqual(t, 10, mockSeries.Len())
|
|
|
|
mas := &SMASeries{
|
|
InnerSeries: mockSeries,
|
|
Period: 10,
|
|
}
|
|
|
|
var yvalues []float64
|
|
for x := 0; x < mas.Len(); x++ {
|
|
_, y := mas.GetValues(x)
|
|
yvalues = append(yvalues, y)
|
|
}
|
|
|
|
testutil.AssertEqual(t, 10.0, yvalues[0])
|
|
testutil.AssertEqual(t, 9.5, yvalues[1])
|
|
testutil.AssertEqual(t, 9.0, yvalues[2])
|
|
testutil.AssertEqual(t, 8.5, yvalues[3])
|
|
testutil.AssertEqual(t, 8.0, yvalues[4])
|
|
testutil.AssertEqual(t, 7.5, yvalues[5])
|
|
testutil.AssertEqual(t, 7.0, yvalues[6])
|
|
testutil.AssertEqual(t, 6.5, yvalues[7])
|
|
testutil.AssertEqual(t, 6.0, yvalues[8])
|
|
}
|
|
|
|
func TestSMASeriesGetLastValueWindowOverlap(t *testing.T) {
|
|
// replaced new assertions helper
|
|
|
|
mockSeries := mockValuesProvider{
|
|
LinearRange(1.0, 10.0),
|
|
LinearRange(10, 1.0),
|
|
}
|
|
testutil.AssertEqual(t, 10, mockSeries.Len())
|
|
|
|
mas := &SMASeries{
|
|
InnerSeries: mockSeries,
|
|
Period: 15,
|
|
}
|
|
|
|
var yvalues []float64
|
|
for x := 0; x < mas.Len(); x++ {
|
|
_, y := mas.GetValues(x)
|
|
yvalues = append(yvalues, y)
|
|
}
|
|
|
|
lx, ly := mas.GetLastValues()
|
|
testutil.AssertEqual(t, 10.0, lx)
|
|
testutil.AssertEqual(t, 5.5, ly)
|
|
testutil.AssertEqual(t, yvalues[len(yvalues)-1], ly)
|
|
}
|
|
|
|
func TestSMASeriesGetLastValue(t *testing.T) {
|
|
// replaced new assertions helper
|
|
|
|
mockSeries := mockValuesProvider{
|
|
LinearRange(1.0, 100.0),
|
|
LinearRange(100, 1.0),
|
|
}
|
|
testutil.AssertEqual(t, 100, mockSeries.Len())
|
|
|
|
mas := &SMASeries{
|
|
InnerSeries: mockSeries,
|
|
Period: 10,
|
|
}
|
|
|
|
var yvalues []float64
|
|
for x := 0; x < mas.Len(); x++ {
|
|
_, y := mas.GetValues(x)
|
|
yvalues = append(yvalues, y)
|
|
}
|
|
|
|
lx, ly := mas.GetLastValues()
|
|
testutil.AssertEqual(t, 100.0, lx)
|
|
testutil.AssertEqual(t, 6, ly)
|
|
testutil.AssertEqual(t, yvalues[len(yvalues)-1], ly)
|
|
}
|