go-chart/linear_regression_series_test.go

package chart

import (
	"testing"

	assert "github.com/blendlabs/go-assert"
)

func TestLinearRegressionSeries(t *testing.T) {
	assert := assert.New(t)

	mainSeries := ContinuousSeries{
		Name:    "A test series",
		XValues: Seq(1.0, 100.0),
		YValues: Seq(1.0, 100.0),
	}

	linRegSeries := &LinearRegressionSeries{
		InnerSeries: mainSeries,
	}

	lrx0, lry0 := linRegSeries.GetValue(0)
	assert.InDelta(1.0, lrx0, 0.0000001)
	assert.InDelta(1.0, lry0, 0.0000001)

	lrxn, lryn := linRegSeries.GetLastValue()
	assert.InDelta(100.0, lrxn, 0.0000001)
	assert.InDelta(100.0, lryn, 0.0000001)
}

func TestLinearRegressionSeriesDesc(t *testing.T) {
	assert := assert.New(t)

	mainSeries := ContinuousSeries{
		Name:    "A test series",
		XValues: Seq(100.0, 1.0),
		YValues: Seq(100.0, 1.0),
	}

	linRegSeries := &LinearRegressionSeries{
		InnerSeries: mainSeries,
	}

	lrx0, lry0 := linRegSeries.GetValue(0)
	assert.InDelta(100.0, lrx0, 0.0000001)
	assert.InDelta(100.0, lry0, 0.0000001)

	lrxn, lryn := linRegSeries.GetLastValue()
	assert.InDelta(1.0, lrxn, 0.0000001)
	assert.InDelta(1.0, lryn, 0.0000001)
}

func TestLinearRegressionSeriesWindowAndOffset(t *testing.T) {
	assert := assert.New(t)

	mainSeries := ContinuousSeries{
		Name:    "A test series",
		XValues: Seq(100.0, 1.0),
		YValues: Seq(100.0, 1.0),
	}

	linRegSeries := &LinearRegressionSeries{
		InnerSeries: mainSeries,
		Offset:      10,
		Window:      10,
	}

	assert.Equal(10, linRegSeries.Len())

	lrx0, lry0 := linRegSeries.GetValue(0)
	assert.InDelta(90.0, lrx0, 0.0000001)
	assert.InDelta(90.0, lry0, 0.0000001)

	lrxn, lryn := linRegSeries.GetLastValue()
	assert.InDelta(80.0, lrxn, 0.0000001)
	assert.InDelta(80.0, lryn, 0.0000001)
}