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: Sequence.Float64(1.0, 100.0), YValues: Sequence.Float64(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: Sequence.Float64(100.0, 1.0), YValues: Sequence.Float64(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: Sequence.Float64(100.0, 1.0), YValues: Sequence.Float64(100.0, 1.0), } linRegSeries := &LinearRegressionSeries{ InnerSeries: mainSeries, Offset: 10, Limit: 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) }