tests.

linear_regression_series_test.go

@@ -0,0 +1,75 @@
+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,
+	}
+
+	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)
+}