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

grid2: improve number func

Browse Source
This commit is contained in:
c9s 2022-11-08 20:23:25 +08:00
parent f46fc7ee80
commit 4fb2230e5d
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pkg/strategy/grid2/grid_test.go
View File

@ -9,19 +9,25 @@ import (
)
func number(a interface{}) fixedpoint.Value {
if s, ok := a.(string); ok {
return fixedpoint.MustNewFromString(s)
switch v := a.(type) {
case string:
return fixedpoint.MustNewFromString(v)
case int:
return fixedpoint.NewFromInt(int64(v))
case int64:
return fixedpoint.NewFromInt(int64(v))
case float64:
return fixedpoint.NewFromFloat(v)
}
f := a.(float64)
return fixedpoint.NewFromFloat(f)
return fixedpoint.Zero
}
func TestNewGrid(t *testing.T) {
upper := fixedpoint.NewFromFloat(500.0)
lower := fixedpoint.NewFromFloat(100.0)
size := fixedpoint.NewFromFloat(100.0)
grid := NewGrid(lower, upper, size, number(2.0))
grid := NewGrid(lower, upper, size, number(0.01))
assert.Equal(t, upper, grid.UpperPrice)
assert.Equal(t, lower, grid.LowerPrice)
assert.Equal(t, fixedpoint.NewFromFloat(4), grid.Spread)
@ -35,7 +41,7 @@ func TestGrid_HasPin(t *testing.T) {
upper := fixedpoint.NewFromFloat(500.0)
lower := fixedpoint.NewFromFloat(100.0)
size := fixedpoint.NewFromFloat(100.0)
grid := NewGrid(lower, upper, size, number(2))
grid := NewGrid(lower, upper, size, number(0.01))
assert.True(t, grid.HasPin(Pin(number(100.0))))
assert.True(t, grid.HasPin(Pin(number(500.0))))