mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-31 15:10:42 +00:00
* move limits, transition to key gen * rollout NewExchangePairAssetKey everywhere * test improvements * self-review fixes * ok, lets go * fix merge issue * slower value func,assertify,drop IsValidPairString * remove binance reference for backtesting test * Redundant nil checks removed due to redundancy * Update order_test.go * Move limits back into /exchanges/ * puts limits in a different box again * SHAZBERT SPECIAL SUGGESTIONS * Update gateio_wrapper.go * fixes all build issues * Many niteroos! * something has gone awry * bugfix * gk's everywhere nits * lint * extra lint * re-remove IsValidPairString * lint fix * standardise test * revert some bads * dupe rm * another revert 360 mcgee * un-in-revertify * Update exchange/order/limits/levels_test.go Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io> * fix * Update exchanges/binance/binance_test.go HERE'S HOPING GITHUB FORMATS THIS CORRECTLY! Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com> * update text * rn func, same line err gk4202000 --------- Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io> Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>
887 lines
21 KiB
Go
887 lines
21 KiB
Go
package statistics
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/shopspring/decimal"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/common"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/data"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventhandlers/portfolio"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventhandlers/portfolio/compliance"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventhandlers/portfolio/holdings"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventtypes/event"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventtypes/fill"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventtypes/kline"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventtypes/order"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventtypes/signal"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/funding"
|
|
gctcommon "github.com/thrasher-corp/gocryptotrader/common"
|
|
"github.com/thrasher-corp/gocryptotrader/common/key"
|
|
"github.com/thrasher-corp/gocryptotrader/currency"
|
|
"github.com/thrasher-corp/gocryptotrader/engine"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
|
|
gctkline "github.com/thrasher-corp/gocryptotrader/exchanges/kline"
|
|
gctorder "github.com/thrasher-corp/gocryptotrader/exchanges/order"
|
|
)
|
|
|
|
const testExchange = "binance"
|
|
|
|
var (
|
|
eleeg = decimal.NewFromInt(1336)
|
|
eleet = decimal.NewFromInt(1337)
|
|
eleeet = decimal.NewFromInt(13337)
|
|
eleeb = decimal.NewFromInt(1338)
|
|
)
|
|
|
|
func TestReset(t *testing.T) {
|
|
t.Parallel()
|
|
s := &Statistic{
|
|
TotalOrders: 1,
|
|
}
|
|
err := s.Reset()
|
|
assert.NoError(t, err)
|
|
|
|
if s.TotalOrders != 0 {
|
|
t.Error("expected 0")
|
|
}
|
|
|
|
s = nil
|
|
err = s.Reset()
|
|
assert.ErrorIs(t, err, gctcommon.ErrNilPointer)
|
|
}
|
|
|
|
func TestAddDataEventForTime(t *testing.T) {
|
|
t.Parallel()
|
|
tt := time.Now()
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
s := Statistic{}
|
|
err := s.SetEventForOffset(nil)
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
},
|
|
Open: eleet,
|
|
Close: eleet,
|
|
Low: eleet,
|
|
High: eleet,
|
|
Volume: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
if s.ExchangeAssetPairStatistics == nil {
|
|
t.Error("expected not nil")
|
|
}
|
|
if len(s.ExchangeAssetPairStatistics[key.NewExchangeAssetPair(exch, a, p)].Events) != 1 {
|
|
t.Error("expected 1 event")
|
|
}
|
|
}
|
|
|
|
func TestAddSignalEventForTime(t *testing.T) {
|
|
t.Parallel()
|
|
tt := time.Now()
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
s := Statistic{}
|
|
err := s.SetEventForOffset(nil)
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
err = s.SetEventForOffset(&signal.Signal{})
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
s.ExchangeAssetPairStatistics = make(map[key.ExchangeAssetPair]*CurrencyPairStatistic)
|
|
b := &event.Base{}
|
|
err = s.SetEventForOffset(&signal.Signal{
|
|
Base: b,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
b.Exchange = exch
|
|
b.Time = tt
|
|
b.Interval = gctkline.OneDay
|
|
b.CurrencyPair = p
|
|
b.AssetType = a
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: b,
|
|
Open: eleet,
|
|
Close: eleet,
|
|
Low: eleet,
|
|
High: eleet,
|
|
Volume: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&signal.Signal{
|
|
Base: b,
|
|
ClosePrice: eleet,
|
|
Direction: gctorder.Buy,
|
|
})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestAddExchangeEventForTime(t *testing.T) {
|
|
t.Parallel()
|
|
tt := time.Now()
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
s := Statistic{}
|
|
err := s.SetEventForOffset(nil)
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
err = s.SetEventForOffset(&order.Order{})
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
s.ExchangeAssetPairStatistics = make(map[key.ExchangeAssetPair]*CurrencyPairStatistic)
|
|
b := &event.Base{}
|
|
|
|
b.Exchange = exch
|
|
b.Time = tt
|
|
b.Interval = gctkline.OneDay
|
|
b.CurrencyPair = p
|
|
b.AssetType = a
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: b,
|
|
Open: eleet,
|
|
Close: eleet,
|
|
Low: eleet,
|
|
High: eleet,
|
|
Volume: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&order.Order{
|
|
Base: b,
|
|
ID: "elite",
|
|
Direction: gctorder.Buy,
|
|
Status: gctorder.New,
|
|
ClosePrice: eleet,
|
|
Amount: eleet,
|
|
OrderType: gctorder.Stop,
|
|
Leverage: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestAddFillEventForTime(t *testing.T) {
|
|
t.Parallel()
|
|
tt := time.Now()
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
s := Statistic{}
|
|
err := s.SetEventForOffset(nil)
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
err = s.SetEventForOffset(&fill.Fill{})
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
s.ExchangeAssetPairStatistics = make(map[key.ExchangeAssetPair]*CurrencyPairStatistic)
|
|
b := &event.Base{}
|
|
err = s.SetEventForOffset(&fill.Fill{
|
|
Base: b,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
b.Exchange = exch
|
|
b.Time = tt
|
|
b.Interval = gctkline.OneDay
|
|
b.CurrencyPair = p
|
|
b.AssetType = a
|
|
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: b,
|
|
Open: eleet,
|
|
Close: eleet,
|
|
Low: eleet,
|
|
High: eleet,
|
|
Volume: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&fill.Fill{
|
|
Base: b,
|
|
Direction: gctorder.Buy,
|
|
Amount: eleet,
|
|
ClosePrice: eleet,
|
|
VolumeAdjustedPrice: eleet,
|
|
PurchasePrice: eleet,
|
|
ExchangeFee: eleet,
|
|
Slippage: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestAddHoldingsForTime(t *testing.T) {
|
|
t.Parallel()
|
|
tt := time.Now()
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
s := Statistic{}
|
|
err := s.AddHoldingsForTime(&holdings.Holding{})
|
|
assert.ErrorIs(t, err, errExchangeAssetPairStatsUnset)
|
|
|
|
s.ExchangeAssetPairStatistics = make(map[key.ExchangeAssetPair]*CurrencyPairStatistic)
|
|
err = s.AddHoldingsForTime(&holdings.Holding{})
|
|
assert.ErrorIs(t, err, errCurrencyStatisticsUnset)
|
|
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
},
|
|
Open: eleet,
|
|
Close: eleet,
|
|
Low: eleet,
|
|
High: eleet,
|
|
Volume: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.AddHoldingsForTime(&holdings.Holding{
|
|
Pair: p,
|
|
Asset: a,
|
|
Exchange: exch,
|
|
Timestamp: tt,
|
|
QuoteInitialFunds: eleet,
|
|
BaseSize: eleet,
|
|
BaseValue: eleet,
|
|
SoldAmount: eleet,
|
|
BoughtAmount: eleet,
|
|
QuoteSize: eleet,
|
|
TotalValueDifference: eleet,
|
|
ChangeInTotalValuePercent: eleet,
|
|
PositionsValueDifference: eleet,
|
|
TotalValue: eleet,
|
|
TotalFees: eleet,
|
|
TotalValueLostToVolumeSizing: eleet,
|
|
TotalValueLostToSlippage: eleet,
|
|
TotalValueLost: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestAddComplianceSnapshotForTime(t *testing.T) {
|
|
t.Parallel()
|
|
tt := time.Now()
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
s := Statistic{}
|
|
|
|
err := s.AddComplianceSnapshotForTime(nil, nil)
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
err = s.AddComplianceSnapshotForTime(nil, &fill.Fill{})
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
err = s.AddComplianceSnapshotForTime(&compliance.Snapshot{}, &fill.Fill{})
|
|
assert.ErrorIs(t, err, errExchangeAssetPairStatsUnset)
|
|
|
|
s.ExchangeAssetPairStatistics = make(map[key.ExchangeAssetPair]*CurrencyPairStatistic)
|
|
b := &event.Base{}
|
|
err = s.AddComplianceSnapshotForTime(&compliance.Snapshot{}, &fill.Fill{Base: b})
|
|
assert.ErrorIs(t, err, errCurrencyStatisticsUnset)
|
|
|
|
b.Exchange = exch
|
|
b.Time = tt
|
|
b.Interval = gctkline.OneDay
|
|
b.CurrencyPair = p
|
|
b.AssetType = a
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: b,
|
|
Open: eleet,
|
|
Close: eleet,
|
|
Low: eleet,
|
|
High: eleet,
|
|
Volume: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.AddComplianceSnapshotForTime(&compliance.Snapshot{
|
|
Timestamp: tt,
|
|
}, &fill.Fill{
|
|
Base: b,
|
|
})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestSerialise(t *testing.T) {
|
|
t.Parallel()
|
|
s := Statistic{}
|
|
if _, err := s.Serialise(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestSetStrategyName(t *testing.T) {
|
|
t.Parallel()
|
|
s := Statistic{}
|
|
s.SetStrategyName("test")
|
|
if s.StrategyName != "test" {
|
|
t.Error("expected test")
|
|
}
|
|
}
|
|
|
|
func TestPrintTotalResults(t *testing.T) {
|
|
t.Parallel()
|
|
s := Statistic{
|
|
FundingStatistics: &FundingStatistics{},
|
|
}
|
|
s.BiggestDrawdown = s.GetTheBiggestDrawdownAcrossCurrencies([]FinalResultsHolder{
|
|
{
|
|
Exchange: "test",
|
|
MaxDrawdown: Swing{
|
|
DrawdownPercent: eleet,
|
|
},
|
|
},
|
|
})
|
|
s.BestStrategyResults = s.GetBestStrategyPerformer([]FinalResultsHolder{
|
|
{
|
|
Exchange: "test",
|
|
Asset: asset.Spot,
|
|
Pair: currency.NewPair(currency.BTC, currency.DOGE),
|
|
MaxDrawdown: Swing{},
|
|
MarketMovement: eleet,
|
|
StrategyMovement: eleet,
|
|
},
|
|
})
|
|
s.BestMarketMovement = s.GetBestMarketPerformer([]FinalResultsHolder{
|
|
{
|
|
Exchange: "test",
|
|
MarketMovement: eleet,
|
|
},
|
|
})
|
|
s.PrintTotalResults()
|
|
}
|
|
|
|
func TestGetBestStrategyPerformer(t *testing.T) {
|
|
t.Parallel()
|
|
s := Statistic{}
|
|
resp := s.GetBestStrategyPerformer(nil)
|
|
if resp.Exchange != "" {
|
|
t.Error("expected unset details")
|
|
}
|
|
|
|
resp = s.GetBestStrategyPerformer([]FinalResultsHolder{
|
|
{
|
|
Exchange: "test",
|
|
Asset: asset.Spot,
|
|
Pair: currency.NewPair(currency.BTC, currency.DOGE),
|
|
MaxDrawdown: Swing{},
|
|
MarketMovement: eleet,
|
|
StrategyMovement: eleet,
|
|
},
|
|
{
|
|
Exchange: "test2",
|
|
Asset: asset.Spot,
|
|
Pair: currency.NewPair(currency.BTC, currency.DOGE),
|
|
MaxDrawdown: Swing{},
|
|
MarketMovement: eleeb,
|
|
StrategyMovement: eleeb,
|
|
},
|
|
})
|
|
|
|
if resp.Exchange != "test2" {
|
|
t.Error("expected test2")
|
|
}
|
|
}
|
|
|
|
func TestGetTheBiggestDrawdownAcrossCurrencies(t *testing.T) {
|
|
t.Parallel()
|
|
s := Statistic{}
|
|
result := s.GetTheBiggestDrawdownAcrossCurrencies(nil)
|
|
if result.Exchange != "" {
|
|
t.Error("expected empty")
|
|
}
|
|
|
|
result = s.GetTheBiggestDrawdownAcrossCurrencies([]FinalResultsHolder{
|
|
{
|
|
Exchange: "test",
|
|
MaxDrawdown: Swing{
|
|
DrawdownPercent: eleet,
|
|
},
|
|
},
|
|
{
|
|
Exchange: "test2",
|
|
MaxDrawdown: Swing{
|
|
DrawdownPercent: eleeb,
|
|
},
|
|
},
|
|
})
|
|
if result.Exchange != "test2" {
|
|
t.Error("expected test2")
|
|
}
|
|
}
|
|
|
|
func TestGetBestMarketPerformer(t *testing.T) {
|
|
t.Parallel()
|
|
s := Statistic{}
|
|
result := s.GetBestMarketPerformer(nil)
|
|
if result.Exchange != "" {
|
|
t.Error("expected empty")
|
|
}
|
|
|
|
result = s.GetBestMarketPerformer([]FinalResultsHolder{
|
|
{
|
|
Exchange: "test",
|
|
MarketMovement: eleet,
|
|
},
|
|
{
|
|
Exchange: "test2",
|
|
MarketMovement: eleeg,
|
|
},
|
|
})
|
|
if result.Exchange != "test" {
|
|
t.Error("expected test")
|
|
}
|
|
}
|
|
|
|
func TestPrintAllEventsChronologically(t *testing.T) {
|
|
t.Parallel()
|
|
s := Statistic{}
|
|
s.PrintAllEventsChronologically()
|
|
tt := time.Now()
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
err := s.SetEventForOffset(nil)
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
},
|
|
Open: eleet,
|
|
Close: eleet,
|
|
Low: eleet,
|
|
High: eleet,
|
|
Volume: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&fill.Fill{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
},
|
|
Direction: gctorder.Buy,
|
|
Amount: eleet,
|
|
ClosePrice: eleet,
|
|
VolumeAdjustedPrice: eleet,
|
|
PurchasePrice: eleet,
|
|
ExchangeFee: eleet,
|
|
Slippage: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&signal.Signal{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
},
|
|
ClosePrice: eleet,
|
|
Direction: gctorder.Buy,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
s.PrintAllEventsChronologically()
|
|
}
|
|
|
|
func TestCalculateTheResults(t *testing.T) {
|
|
t.Parallel()
|
|
s := Statistic{}
|
|
err := s.CalculateAllResults()
|
|
assert.ErrorIs(t, err, gctcommon.ErrNilPointer)
|
|
|
|
tt := time.Now().Add(-gctkline.OneDay.Duration() * 7)
|
|
tt2 := time.Now().Add(-gctkline.OneDay.Duration() * 6)
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
p2 := currency.NewPair(currency.XRP, currency.DOGE)
|
|
err = s.SetEventForOffset(nil)
|
|
assert.ErrorIs(t, err, common.ErrNilEvent)
|
|
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
Offset: 1,
|
|
},
|
|
Open: eleet,
|
|
Close: eleet,
|
|
Low: eleet,
|
|
High: eleet,
|
|
Volume: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&signal.Signal{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
Offset: 1,
|
|
},
|
|
OpenPrice: eleet,
|
|
HighPrice: eleet,
|
|
LowPrice: eleet,
|
|
ClosePrice: eleet,
|
|
Volume: eleet,
|
|
Direction: gctorder.Buy,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p2,
|
|
AssetType: a,
|
|
Offset: 2,
|
|
},
|
|
Open: eleeb,
|
|
Close: eleeb,
|
|
Low: eleeb,
|
|
High: eleeb,
|
|
Volume: eleeb,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&signal.Signal{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p2,
|
|
AssetType: a,
|
|
Offset: 2,
|
|
},
|
|
OpenPrice: eleet,
|
|
HighPrice: eleet,
|
|
LowPrice: eleet,
|
|
ClosePrice: eleet,
|
|
Volume: eleet,
|
|
Direction: gctorder.Buy,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt2,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
Offset: 3,
|
|
},
|
|
Open: eleeb,
|
|
Close: eleeb,
|
|
Low: eleeb,
|
|
High: eleeb,
|
|
Volume: eleeb,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&signal.Signal{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt2,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
Offset: 3,
|
|
},
|
|
OpenPrice: eleeb,
|
|
HighPrice: eleeb,
|
|
LowPrice: eleeb,
|
|
ClosePrice: eleeb,
|
|
Volume: eleeb,
|
|
Direction: gctorder.Buy,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt2,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p2,
|
|
AssetType: a,
|
|
Offset: 4,
|
|
},
|
|
Open: eleeb,
|
|
Close: eleeb,
|
|
Low: eleeb,
|
|
High: eleeb,
|
|
Volume: eleeb,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
signal4 := &signal.Signal{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt2,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p2,
|
|
AssetType: a,
|
|
Offset: 4,
|
|
},
|
|
OpenPrice: eleeb,
|
|
HighPrice: eleeb,
|
|
LowPrice: eleeb,
|
|
ClosePrice: eleeb,
|
|
Volume: eleeb,
|
|
Direction: gctorder.Buy,
|
|
}
|
|
err = s.SetEventForOffset(signal4)
|
|
assert.NoError(t, err)
|
|
|
|
mapKey1 := key.NewExchangeAssetPair(exch, a, p)
|
|
mapKey2 := key.NewExchangeAssetPair(exch, a, p2)
|
|
s.ExchangeAssetPairStatistics[mapKey1].Events[1].Holdings.QuoteInitialFunds = eleet
|
|
s.ExchangeAssetPairStatistics[mapKey1].Events[1].Holdings.TotalValue = eleeet
|
|
s.ExchangeAssetPairStatistics[mapKey2].Events[1].Holdings.QuoteInitialFunds = eleet
|
|
s.ExchangeAssetPairStatistics[mapKey2].Events[1].Holdings.TotalValue = eleeet
|
|
|
|
funds, err := funding.SetupFundingManager(&engine.ExchangeManager{}, false, false, false)
|
|
assert.NoError(t, err)
|
|
|
|
pBase, err := funding.CreateItem(exch, a, p.Base, eleeet, decimal.Zero)
|
|
assert.NoError(t, err)
|
|
|
|
pQuote, err := funding.CreateItem(exch, a, p.Quote, eleeet, decimal.Zero)
|
|
assert.NoError(t, err)
|
|
|
|
pair, err := funding.CreatePair(pBase, pQuote)
|
|
assert.NoError(t, err)
|
|
|
|
err = funds.AddPair(pair)
|
|
assert.NoError(t, err)
|
|
|
|
pBase2, err := funding.CreateItem(exch, a, p2.Base, eleeet, decimal.Zero)
|
|
assert.NoError(t, err)
|
|
|
|
pQuote2, err := funding.CreateItem(exch, a, p2.Quote, eleeet, decimal.Zero)
|
|
assert.NoError(t, err)
|
|
|
|
pair2, err := funding.CreatePair(pBase2, pQuote2)
|
|
assert.NoError(t, err)
|
|
|
|
err = funds.AddPair(pair2)
|
|
assert.NoError(t, err)
|
|
|
|
s.FundManager = funds
|
|
err = s.CalculateAllResults()
|
|
assert.ErrorIs(t, err, errMissingSnapshots)
|
|
|
|
err = s.CalculateAllResults()
|
|
assert.ErrorIs(t, err, errMissingSnapshots)
|
|
|
|
funds, err = funding.SetupFundingManager(&engine.ExchangeManager{}, false, true, false)
|
|
assert.NoError(t, err)
|
|
|
|
err = funds.AddPair(pair)
|
|
assert.NoError(t, err)
|
|
|
|
err = funds.AddPair(pair2)
|
|
assert.NoError(t, err)
|
|
|
|
s.FundManager = funds
|
|
err = s.CalculateAllResults()
|
|
assert.ErrorIs(t, err, errMissingSnapshots)
|
|
|
|
err = s.AddComplianceSnapshotForTime(&compliance.Snapshot{Timestamp: tt2}, signal4)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestCalculateBiggestEventDrawdown(t *testing.T) {
|
|
tt1 := time.Now().Add(-gctkline.OneDay.Duration() * 7).Round(gctkline.OneDay.Duration())
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
var events []data.Event
|
|
for i := range int64(100) {
|
|
tt1 = tt1.Add(gctkline.OneDay.Duration())
|
|
even := &event.Base{
|
|
Exchange: exch,
|
|
Time: tt1,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
}
|
|
if i == 50 {
|
|
// throw in a wrench, a spike in price
|
|
events = append(events, &kline.Kline{
|
|
Base: even,
|
|
Close: decimal.NewFromInt(1336),
|
|
High: decimal.NewFromInt(1336),
|
|
Low: decimal.NewFromInt(1336),
|
|
})
|
|
} else {
|
|
events = append(events, &kline.Kline{
|
|
Base: even,
|
|
Close: decimal.NewFromInt(1337).Sub(decimal.NewFromInt(i)),
|
|
High: decimal.NewFromInt(1337).Sub(decimal.NewFromInt(i)),
|
|
Low: decimal.NewFromInt(1337).Sub(decimal.NewFromInt(i)),
|
|
})
|
|
}
|
|
}
|
|
|
|
tt1 = tt1.Add(gctkline.OneDay.Duration())
|
|
even := &event.Base{
|
|
Exchange: exch,
|
|
Time: tt1,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
}
|
|
events = append(events, &kline.Kline{
|
|
Base: even,
|
|
Close: decimal.NewFromInt(1338),
|
|
High: decimal.NewFromInt(1338),
|
|
Low: decimal.NewFromInt(1338),
|
|
})
|
|
|
|
tt1 = tt1.Add(gctkline.OneDay.Duration())
|
|
even = &event.Base{
|
|
Exchange: exch,
|
|
Time: tt1,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
}
|
|
events = append(events, &kline.Kline{
|
|
Base: even,
|
|
Close: decimal.NewFromInt(1337),
|
|
High: decimal.NewFromInt(1337),
|
|
Low: decimal.NewFromInt(1337),
|
|
})
|
|
|
|
tt1 = tt1.Add(gctkline.OneDay.Duration())
|
|
even = &event.Base{
|
|
Exchange: exch,
|
|
Time: tt1,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
}
|
|
events = append(events, &kline.Kline{
|
|
Base: even,
|
|
Close: decimal.NewFromInt(1339),
|
|
High: decimal.NewFromInt(1339),
|
|
Low: decimal.NewFromInt(1339),
|
|
})
|
|
|
|
_, err := CalculateBiggestEventDrawdown(nil)
|
|
assert.ErrorIs(t, err, errReceivedNoData)
|
|
|
|
resp, err := CalculateBiggestEventDrawdown(events)
|
|
assert.NoError(t, err)
|
|
|
|
if resp.Highest.Value != decimal.NewFromInt(1337) && !resp.Lowest.Value.Equal(decimal.NewFromInt(1238)) {
|
|
t.Error("unexpected max drawdown")
|
|
}
|
|
|
|
// bogus scenario
|
|
bogusEvent := []data.Event{
|
|
&kline.Kline{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
},
|
|
Close: decimal.NewFromInt(1339),
|
|
High: decimal.NewFromInt(1339),
|
|
Low: decimal.NewFromInt(1339),
|
|
},
|
|
}
|
|
_, err = CalculateBiggestEventDrawdown(bogusEvent)
|
|
assert.ErrorIs(t, err, gctcommon.ErrDateUnset)
|
|
}
|
|
|
|
func TestCalculateBiggestValueAtTimeDrawdown(t *testing.T) {
|
|
var interval gctkline.Interval
|
|
_, err := CalculateBiggestValueAtTimeDrawdown(nil, interval)
|
|
assert.ErrorIs(t, err, errReceivedNoData)
|
|
|
|
_, err = CalculateBiggestValueAtTimeDrawdown(nil, interval)
|
|
assert.ErrorIs(t, err, errReceivedNoData)
|
|
}
|
|
|
|
func TestAddPNLForTime(t *testing.T) {
|
|
t.Parallel()
|
|
s := &Statistic{}
|
|
err := s.AddPNLForTime(nil)
|
|
assert.ErrorIs(t, err, gctcommon.ErrNilPointer)
|
|
|
|
sum := &portfolio.PNLSummary{}
|
|
err = s.AddPNLForTime(sum)
|
|
assert.ErrorIs(t, err, errExchangeAssetPairStatsUnset)
|
|
|
|
tt := time.Now().Add(-gctkline.OneDay.Duration() * 7)
|
|
exch := testExchange
|
|
a := asset.Spot
|
|
p := currency.NewBTCUSDT()
|
|
err = s.SetEventForOffset(&kline.Kline{
|
|
Base: &event.Base{
|
|
Exchange: exch,
|
|
Time: tt,
|
|
Interval: gctkline.OneDay,
|
|
CurrencyPair: p,
|
|
AssetType: a,
|
|
Offset: 1,
|
|
},
|
|
Open: eleet,
|
|
Close: eleet,
|
|
Low: eleet,
|
|
High: eleet,
|
|
Volume: eleet,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = s.AddPNLForTime(sum)
|
|
assert.ErrorIs(t, err, errCurrencyStatisticsUnset)
|
|
|
|
sum.Exchange = exch
|
|
sum.Asset = a
|
|
sum.Pair = p
|
|
err = s.AddPNLForTime(sum)
|
|
assert.ErrorIs(t, err, errNoDataAtOffset)
|
|
|
|
sum.Offset = 1
|
|
err = s.AddPNLForTime(sum)
|
|
assert.NoError(t, err)
|
|
}
|