a211e88530
* updates * updates * tests. * test coverage * fixing test * stride not rows + cols * lu decomp implementation. * poly regression! * poly regression works. * typo.
78 lines
1.7 KiB
Go
78 lines
1.7 KiB
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: 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)
|
|
}
|