Files
gocryptotrader/backtester/eventhandlers/statistics/statistics_test.go
Scott 017cdf1384 Backtester: Live trading upgrades (#1023)
* Modifications for a smoother live run

* Fixes data appending

* Successfully allows multi-currency live trading. Adds multiple currencies to live DCA strategy

* Attempting to get cash and carry working

* Poor attempts at sorting out data and appending it properly with USD in mind

* =designs new live data handler

* Updates cash and carry strat to work

* adds test coverage. begins closeallpositions function

* Updates cash and carry to work live

* New kline.Event type. Cancels orders on close. Rn types

* =Fixes USD funding issue

* =fixes tests

* fixes tests AGAIN

* adds coverage to close all orders

* crummy tests, should override

* more tests

* more tests

* more coverage

* removes scourge of currency.Pair maps. More tests

* missed currency stuff

* Fixes USD data issue & collateral issue. Needs to close ALL orders

* Now triggers updates on the very first data entry

* All my problems are solved now????

* fixes tests, extends coverage

* there is some really funky candle stuff going on

* my brain is melting

* better shutdown management, fixes freezing bug

* fixes data duplication issues, adds retries to requests

* reduces logging, adds verbose options

* expands coverage over all new functionality

* fixes fun bug from curr == curr to curr.Equal(curr)

* fixes setup issues and tests

* starts adding external wallet amounts for funding

* more setup for assets

* setup live fund calcs and placing orders

* successfully performs automated cash and carry

* merge fixes

* funding properly set at all times

* fixes some bugs, need to address currencystatistics still

* adds 'appeneded' trait, attempts to fix some stats

* fixes stat bugs, adds cool new fetchfees feature

* fixes terrible processing bugs

* tightens realorder stats, sadly loses some live stats

* this actually sets everything correctly for bothcd ..cd ..cd ..cd ..cd ..!

* fix tests

* coverage

* beautiful new test coverage

* docs

* adds new fee getter delayer

* commits from the correct directory

* Lint

* adds verbose to fund manager

* Fix bug in t2b2 strat. Update dca live config. Docs

* go mod tidy

* update buf

* buf + test improvement

* Post merge fixes

* fixes surprise offset bug

* fix sizing restrictions for cash and carry

* fix server lints

* merge fixes

* test fixesss

* lintle fixles

* slowloris

* rn run to task, bug fixes, close all on close

* rpc lint and fixes

* bugfix: order manager not processing orders properly

* somewhat addresses nits

* absolutely broken end of day commit

* absolutely massive knockon effects from nits

* massive knockon effects continue

* fixes things

* address remaining nits

* jk now fixes things

* addresses the easier nits

* more nit fixers

* more niterinos addressederinos

* refactors holdings and does some nits

* so buf

* addresses some nits, fixes holdings bugs

* cleanup

* attempts to fix alert chans to prevent many chans waiting?

* terrible code, will revert

* to be reviewed in detail tomorrow

* Fixes up channel system

* smashes those nits

* fixes extra candles, fixes collateral bug, tests

* fixes data races, introduces reflection

* more checks n tests

* Fixes cash and carry issues. Fixes more cool bugs

* fixes ~typer~ typo

* replace spot strats from ftx to binance

* fixes all the tests I just destroyed

* removes example path, rm verbose

* 1) what 2) removes FTX references from the Backtester

* renamed, non-working strategies

* Removes FTX references almost as fast as sbf removes funds

* regen docs, add contrib names,sort contrib names

* fixes merge renamings

* Addresses nits. Fixes setting API credentials. Fixes Binance limit retrieval

* Fixes live order bugs with real orders and without

* Apply suggestions from code review

Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io>

* Update backtester/engine/live.go

Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io>

* Update backtester/engine/live.go

Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io>

* Update backtester/config/strategyconfigbuilder/main.go

Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io>

* updates docs

* even better docs

Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io>
2023-01-05 13:03:17 +11:00

984 lines
25 KiB
Go

package statistics
import (
"errors"
"testing"
"time"
"github.com/shopspring/decimal"
"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/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()
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
if s.TotalOrders != 0 {
t.Error("expected 0")
}
s = nil
err = s.Reset()
if !errors.Is(err, gctcommon.ErrNilPointer) {
t.Errorf("received: %v, expected: %v", err, gctcommon.ErrNilPointer)
}
}
func TestAddDataEventForTime(t *testing.T) {
t.Parallel()
tt := time.Now()
exch := testExchange
a := asset.Spot
p := currency.NewPair(currency.BTC, currency.USDT)
s := Statistic{}
err := s.SetEventForOffset(nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", 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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
if s.ExchangeAssetPairStatistics == nil {
t.Error("expected not nil")
}
if len(s.ExchangeAssetPairStatistics[exch][a][p.Base.Item][p.Quote.Item].Events) != 1 {
t.Error("expected 1 event")
}
}
func TestAddSignalEventForTime(t *testing.T) {
t.Parallel()
tt := time.Now()
exch := testExchange
a := asset.Spot
p := currency.NewPair(currency.BTC, currency.USDT)
s := Statistic{}
err := s.SetEventForOffset(nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
err = s.SetEventForOffset(&signal.Signal{})
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
s.ExchangeAssetPairStatistics = make(map[string]map[asset.Item]map[*currency.Item]map[*currency.Item]*CurrencyPairStatistic)
b := &event.Base{}
err = s.SetEventForOffset(&signal.Signal{
Base: b,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = s.SetEventForOffset(&signal.Signal{
Base: b,
ClosePrice: eleet,
Direction: gctorder.Buy,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
}
func TestAddExchangeEventForTime(t *testing.T) {
t.Parallel()
tt := time.Now()
exch := testExchange
a := asset.Spot
p := currency.NewPair(currency.BTC, currency.USDT)
s := Statistic{}
err := s.SetEventForOffset(nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
err = s.SetEventForOffset(&order.Order{})
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
s.ExchangeAssetPairStatistics = make(map[string]map[asset.Item]map[*currency.Item]map[*currency.Item]*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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = s.SetEventForOffset(&order.Order{
Base: b,
ID: "elite",
Direction: gctorder.Buy,
Status: gctorder.New,
ClosePrice: eleet,
Amount: eleet,
OrderType: gctorder.Stop,
Leverage: eleet,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
}
func TestAddFillEventForTime(t *testing.T) {
t.Parallel()
tt := time.Now()
exch := testExchange
a := asset.Spot
p := currency.NewPair(currency.BTC, currency.USDT)
s := Statistic{}
err := s.SetEventForOffset(nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
err = s.SetEventForOffset(&fill.Fill{})
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
s.ExchangeAssetPairStatistics = make(map[string]map[asset.Item]map[*currency.Item]map[*currency.Item]*CurrencyPairStatistic)
b := &event.Base{}
err = s.SetEventForOffset(&fill.Fill{
Base: b,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = s.SetEventForOffset(&fill.Fill{
Base: b,
Direction: gctorder.Buy,
Amount: eleet,
ClosePrice: eleet,
VolumeAdjustedPrice: eleet,
PurchasePrice: eleet,
ExchangeFee: eleet,
Slippage: eleet,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
}
func TestAddHoldingsForTime(t *testing.T) {
t.Parallel()
tt := time.Now()
exch := testExchange
a := asset.Spot
p := currency.NewPair(currency.BTC, currency.USDT)
s := Statistic{}
err := s.AddHoldingsForTime(&holdings.Holding{})
if !errors.Is(err, errExchangeAssetPairStatsUnset) {
t.Errorf("received: %v, expected: %v", err, errExchangeAssetPairStatsUnset)
}
s.ExchangeAssetPairStatistics = make(map[string]map[asset.Item]map[*currency.Item]map[*currency.Item]*CurrencyPairStatistic)
err = s.AddHoldingsForTime(&holdings.Holding{})
if !errors.Is(err, errCurrencyStatisticsUnset) {
t.Errorf("received: %v, expected: %v", 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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
}
func TestAddComplianceSnapshotForTime(t *testing.T) {
t.Parallel()
tt := time.Now()
exch := testExchange
a := asset.Spot
p := currency.NewPair(currency.BTC, currency.USDT)
s := Statistic{}
err := s.AddComplianceSnapshotForTime(nil, nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
err = s.AddComplianceSnapshotForTime(nil, &fill.Fill{})
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
err = s.AddComplianceSnapshotForTime(&compliance.Snapshot{}, &fill.Fill{})
if !errors.Is(err, errExchangeAssetPairStatsUnset) {
t.Errorf("received: %v, expected: %v", err, errExchangeAssetPairStatsUnset)
}
s.ExchangeAssetPairStatistics = make(map[string]map[asset.Item]map[*currency.Item]map[*currency.Item]*CurrencyPairStatistic)
b := &event.Base{}
err = s.AddComplianceSnapshotForTime(&compliance.Snapshot{}, &fill.Fill{Base: b})
if !errors.Is(err, errCurrencyStatisticsUnset) {
t.Errorf("received: %v, expected: %v", 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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = s.AddComplianceSnapshotForTime(&compliance.Snapshot{
Timestamp: tt,
}, &fill.Fill{
Base: b,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
}
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.NewPair(currency.BTC, currency.USDT)
err := s.SetEventForOffset(nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", 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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = s.SetEventForOffset(&signal.Signal{
Base: &event.Base{
Exchange: exch,
Time: tt,
Interval: gctkline.OneDay,
CurrencyPair: p,
AssetType: a,
},
ClosePrice: eleet,
Direction: gctorder.Buy,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
s.PrintAllEventsChronologically()
}
func TestCalculateTheResults(t *testing.T) {
t.Parallel()
s := Statistic{}
err := s.CalculateAllResults()
if !errors.Is(err, gctcommon.ErrNilPointer) {
t.Errorf("received: %v, expected: %v", 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.NewPair(currency.BTC, currency.USDT)
p2 := currency.NewPair(currency.XRP, currency.DOGE)
err = s.SetEventForOffset(nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", 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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
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)
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
s.ExchangeAssetPairStatistics[exch][a][p.Base.Item][p.Quote.Item].Events[1].Holdings.QuoteInitialFunds = eleet
s.ExchangeAssetPairStatistics[exch][a][p.Base.Item][p.Quote.Item].Events[1].Holdings.TotalValue = eleeet
s.ExchangeAssetPairStatistics[exch][a][p2.Base.Item][p2.Quote.Item].Events[1].Holdings.QuoteInitialFunds = eleet
s.ExchangeAssetPairStatistics[exch][a][p2.Base.Item][p2.Quote.Item].Events[1].Holdings.TotalValue = eleeet
funds, err := funding.SetupFundingManager(&engine.ExchangeManager{}, false, false, false)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
pBase, err := funding.CreateItem(exch, a, p.Base, eleeet, decimal.Zero)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
pQuote, err := funding.CreateItem(exch, a, p.Quote, eleeet, decimal.Zero)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
pair, err := funding.CreatePair(pBase, pQuote)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
err = funds.AddPair(pair)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
pBase2, err := funding.CreateItem(exch, a, p2.Base, eleeet, decimal.Zero)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
pQuote2, err := funding.CreateItem(exch, a, p2.Quote, eleeet, decimal.Zero)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
pair2, err := funding.CreatePair(pBase2, pQuote2)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
err = funds.AddPair(pair2)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
s.FundManager = funds
err = s.CalculateAllResults()
if !errors.Is(err, errMissingSnapshots) {
t.Errorf("received '%v' expected '%v'", err, errMissingSnapshots)
}
err = s.CalculateAllResults()
if !errors.Is(err, errMissingSnapshots) {
t.Errorf("received '%v' expected '%v'", err, errMissingSnapshots)
}
funds, err = funding.SetupFundingManager(&engine.ExchangeManager{}, false, true, false)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
err = funds.AddPair(pair)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
err = funds.AddPair(pair2)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
s.FundManager = funds
err = s.CalculateAllResults()
if !errors.Is(err, errMissingSnapshots) {
t.Errorf("received '%v' expected '%v'", err, errMissingSnapshots)
}
err = s.AddComplianceSnapshotForTime(&compliance.Snapshot{Timestamp: tt2}, signal4)
if !errors.Is(err, nil) {
t.Errorf("received '%v' expected '%v'", err, nil)
}
}
func TestCalculateBiggestEventDrawdown(t *testing.T) {
tt1 := time.Now().Add(-gctkline.OneDay.Duration() * 7).Round(gctkline.OneDay.Duration())
exch := testExchange
a := asset.Spot
p := currency.NewPair(currency.BTC, currency.USDT)
var events []data.Event
for i := int64(0); i < 100; i++ {
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)
if !errors.Is(err, errReceivedNoData) {
t.Errorf("received %v expected %v", err, errReceivedNoData)
}
resp, err := CalculateBiggestEventDrawdown(events)
if !errors.Is(err, nil) {
t.Errorf("received %v expected %v", err, nil)
}
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)
if !errors.Is(err, gctcommon.ErrDateUnset) {
t.Errorf("received %v expected %v", err, gctcommon.ErrDateUnset)
}
}
func TestCalculateBiggestValueAtTimeDrawdown(t *testing.T) {
var interval gctkline.Interval
_, err := CalculateBiggestValueAtTimeDrawdown(nil, interval)
if !errors.Is(err, errReceivedNoData) {
t.Errorf("received %v expected %v", err, errReceivedNoData)
}
_, err = CalculateBiggestValueAtTimeDrawdown(nil, interval)
if !errors.Is(err, errReceivedNoData) {
t.Errorf("received %v expected %v", err, errReceivedNoData)
}
}
func TestAddPNLForTime(t *testing.T) {
t.Parallel()
s := &Statistic{}
err := s.AddPNLForTime(nil)
if !errors.Is(err, gctcommon.ErrNilPointer) {
t.Errorf("received %v expected %v", err, gctcommon.ErrNilPointer)
}
sum := &portfolio.PNLSummary{}
err = s.AddPNLForTime(sum)
if !errors.Is(err, errExchangeAssetPairStatsUnset) {
t.Errorf("received %v expected %v", err, errExchangeAssetPairStatsUnset)
}
tt := time.Now().Add(-gctkline.OneDay.Duration() * 7)
exch := testExchange
a := asset.Spot
p := currency.NewPair(currency.BTC, currency.USDT)
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,
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = s.AddPNLForTime(sum)
if !errors.Is(err, errCurrencyStatisticsUnset) {
t.Errorf("received %v expected %v", err, errCurrencyStatisticsUnset)
}
sum.Exchange = exch
sum.Asset = a
sum.Pair = p
err = s.AddPNLForTime(sum)
if !errors.Is(err, errNoDataAtOffset) {
t.Errorf("received %v expected %v", err, errNoDataAtOffset)
}
sum.Offset = 1
err = s.AddPNLForTime(sum)
if !errors.Is(err, nil) {
t.Errorf("received %v expected %v", err, nil)
}
}