31 lines
805 B
Go
31 lines
805 B
Go
|
package indicatorv2
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/fixedpoint"
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_v2_Subtract(t *testing.T) {
|
||
|
|
stream := &types.StandardStream{}
|
||
|
|
kLines := KLines(stream, "", "")
|
||
|
|
closePrices := ClosePrices(kLines)
|
||
|
|
fastEMA := EWMA2(closePrices, 10)
|
||
|
|
slowEMA := EWMA2(closePrices, 25)
|
||
|
|
subtract := Subtract(fastEMA, slowEMA)
|
||
|
|
|
||
|
|
for i := .0; i < 50.0; i++ {
|
||
|
|
stream.EmitKLineClosed(types.KLine{Close: fixedpoint.NewFromFloat(19_000.0 + i)})
|
||
|
|
}
|
||
|
|
|
||
|
|
t.Logf("fastEMA: %+v", fastEMA.Slice)
|
||
|
|
t.Logf("slowEMA: %+v", slowEMA.Slice)
|
||
|
|
|
||
|
|
assert.Equal(t, len(subtract.a), len(subtract.b))
|
||
|
|
assert.Equal(t, len(subtract.a), len(subtract.Slice))
|
||
|
|
assert.InDelta(t, subtract.Slice[0], subtract.a[0]-subtract.b[0], 0.0001)
|
||
|
|
}
|