go-chart/sma_series_test.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)
}