mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-22 14:55:16 +00:00
98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
package xmaker
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/c9s/bbgo/pkg/fixedpoint"
|
|
"github.com/c9s/bbgo/pkg/types"
|
|
|
|
. "github.com/c9s/bbgo/pkg/testing/testhelper"
|
|
)
|
|
|
|
func TestStrategy_getLayerPrice(t *testing.T) {
|
|
symbol := "BTCUSDT"
|
|
market := Market(symbol)
|
|
|
|
s := &Strategy{
|
|
UseDepthPrice: true,
|
|
DepthQuantity: Number(3.0),
|
|
makerMarket: market,
|
|
}
|
|
|
|
sourceBook := types.NewStreamBook(symbol, types.ExchangeBinance)
|
|
sourceBook.Load(types.SliceOrderBook{
|
|
Symbol: symbol,
|
|
Bids: PriceVolumeSlice(
|
|
Number(1300.0), Number(1.0),
|
|
Number(1200.0), Number(2.0),
|
|
Number(1100.0), Number(3.0),
|
|
),
|
|
Asks: PriceVolumeSlice(
|
|
Number(1301.0), Number(1.0),
|
|
Number(1400.0), Number(2.0),
|
|
Number(1500.0), Number(3.0),
|
|
),
|
|
Time: time.Time{},
|
|
LastUpdateId: 1,
|
|
})
|
|
|
|
quote := &Quote{
|
|
BestBidPrice: Number(1300.0),
|
|
BestAskPrice: Number(1301.0),
|
|
BidMargin: Number(0.001),
|
|
AskMargin: Number(0.001),
|
|
BidLayerPips: Number(100.0),
|
|
AskLayerPips: Number(100.0),
|
|
}
|
|
|
|
t.Run("depthPrice bid price at 0", func(t *testing.T) {
|
|
price := s.getLayerPrice(0, types.SideTypeBuy, sourceBook, quote, s.DepthQuantity)
|
|
|
|
// (1300 + 1200*2)/3 * (1 - 0.001)
|
|
assert.InDelta(t, 1232.10, price.Float64(), 0.01)
|
|
})
|
|
|
|
t.Run("depthPrice bid price at 1", func(t *testing.T) {
|
|
price := s.getLayerPrice(1, types.SideTypeBuy, sourceBook, quote, s.DepthQuantity)
|
|
|
|
// (1300 + 1200*2)/3 * (1 - 0.001) - 100 * 0.01
|
|
assert.InDelta(t, 1231.10, price.Float64(), 0.01)
|
|
})
|
|
|
|
t.Run("depthPrice ask price at 0", func(t *testing.T) {
|
|
price := s.getLayerPrice(0, types.SideTypeSell, sourceBook, quote, s.DepthQuantity)
|
|
|
|
// (1301 + 1400*2)/3 * (1 + 0.001)
|
|
assert.InDelta(t, 1368.367, price.Float64(), 0.01)
|
|
})
|
|
|
|
t.Run("depthPrice ask price at 1", func(t *testing.T) {
|
|
price := s.getLayerPrice(1, types.SideTypeSell, sourceBook, quote, s.DepthQuantity)
|
|
|
|
// (1301 + 1400*2)/3 * (1 + 0.001) + 100 * 0.01
|
|
assert.InDelta(t, 1369.367, price.Float64(), 0.01)
|
|
})
|
|
|
|
}
|
|
|
|
func Test_aggregatePrice(t *testing.T) {
|
|
bids := PriceVolumeSliceFromText(`
|
|
1000.0, 1.0
|
|
1200.0, 1.0
|
|
1400.0, 1.0
|
|
`)
|
|
|
|
aggregatedPrice1 := aggregatePrice(bids, fixedpoint.NewFromFloat(0.5))
|
|
assert.Equal(t, fixedpoint.NewFromFloat(1000.0), aggregatedPrice1)
|
|
|
|
aggregatedPrice2 := aggregatePrice(bids, fixedpoint.NewFromInt(1))
|
|
assert.Equal(t, fixedpoint.NewFromFloat(1000.0), aggregatedPrice2)
|
|
|
|
aggregatedPrice3 := aggregatePrice(bids, fixedpoint.NewFromInt(2))
|
|
assert.Equal(t, fixedpoint.NewFromFloat(1100.0), aggregatedPrice3)
|
|
|
|
}
|