lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-27 09:15:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

csvsource: fix TestCSVKLineReader_ReadWithBinanceDecoder test

Browse Source
This commit is contained in:
c9s 2023-12-14 15:56:49 +08:00
parent dae140aacc
commit ee4455fa85
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
2 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/datasource/csvsource/csv_kline_decoder.go
View File

@ -54,37 +54,46 @@ func BinanceCSVKLineDecoder(record []string, interval time.Duration) (types.KLin
k, empty types.KLine
err error
)
if len(record) < 5 {
return k, ErrNotEnoughColumns
}
ts, err := strconv.ParseFloat(record[0], 64) // check for e numbers "1.70027E+12"
if err != nil {
return empty, ErrInvalidTimeFormat
}
open, err := fixedpoint.NewFromString(record[1])
if err != nil {
return empty, ErrInvalidPriceFormat
}
high, err := fixedpoint.NewFromString(record[2])
if err != nil {
return empty, ErrInvalidPriceFormat
}
low, err := fixedpoint.NewFromString(record[3])
if err != nil {
return empty, ErrInvalidPriceFormat
}
closing, err := fixedpoint.NewFromString(record[4])
if err != nil {
return empty, ErrInvalidPriceFormat
}
volume, err := fixedpoint.NewFromString(record[5])
if err != nil {
return empty, ErrInvalidVolumeFormat
volume := fixedpoint.Zero
if len(record) == 6 {
volume, err = fixedpoint.NewFromString(record[5])
if err != nil {
return empty, ErrInvalidVolumeFormat
}
}
k.StartTime = types.NewTimeFromUnix(int64(ts), 0)
k.EndTime = types.NewTimeFromUnix(k.StartTime.Time().Add(interval).Unix(), 0)
k.StartTime = types.Time(time.UnixMilli(int64(ts)))
k.EndTime = types.Time(k.StartTime.Time().Add(interval))
k.Open = open
k.High = high
k.Low = low

2
pkg/datasource/csvsource/csv_kline_reader_test.go
View File

@ -15,7 +15,7 @@ import (
)
func assertKLineEq(t *testing.T, exp, act types.KLine, name string) {
assert.Equal(t, exp.StartTime, act.StartTime, name)
assert.True(t, exp.StartTime.Equal(act.StartTime.Time()), name)
assert.Equal(t, 0, exp.Open.Compare(act.Open), name)
assert.Equal(t, 0, exp.High.Compare(act.High), name)
assert.Equal(t, 0, exp.Low.Compare(act.Low), name)