mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-24 07:45:15 +00:00
38 lines
949 B
Go
38 lines
949 B
Go
|
package indicatorv2
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/csv"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"github.com/c9s/bbgo/pkg/datasource/csvsource"
|
||
|
|
"github.com/c9s/bbgo/pkg/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestVolumeProfile(t *testing.T) {
|
||
|
|
file, _ := os.Open(path.Join("testdata", "BTCUSDT-1m-2022-05-06.csv"))
|
||
|
|
defer func() {
|
||
|
|
assert.NoError(t, file.Close())
|
||
|
|
}()
|
||
|
|
|
||
|
|
candles, err := csvsource.NewCSVKLineReader(csv.NewReader(file)).ReadAll(time.Minute)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
stream := &types.StandardStream{}
|
||
|
|
kLines := KLines(stream, "", "")
|
||
|
|
ind := NewVolumeProfile(kLines, 10)
|
||
|
|
|
||
|
|
for _, candle := range candles {
|
||
|
|
stream.EmitKLineClosed(candle)
|
||
|
|
}
|
||
|
|
assert.InDelta(t, 36512.7, ind.VP.Low, 0.01, "VP.LOW")
|
||
|
|
assert.InDelta(t, 36512.7, ind.VP.VAL, 0.01, "VP.VAL")
|
||
|
|
assert.InDelta(t, 36518.574, ind.VP.POC, 0.01, "VP.POC")
|
||
|
|
assert.InDelta(t, 36530.322, ind.VP.VAH, 0.01, "VP.VAH")
|
||
|
|
assert.InDelta(t, 36617.433, ind.VP.High, 0.01, "VP.HIGH")
|
||
|
|
}
|