Files
gocryptotrader/backtester/eventhandlers/statistics/statistics_test.go
Scott f929b4d51e backtester: Futures handling & FTX Cash and Carry example strategy (#930)
* implements futures functions and GRPC functions on new branch

* lint and test fixes

* Fix uneven split pnl. Adds collateral weight test. docs. New clear func

* Test protection if someone has zero collateral

* Uses string instead of double for accuracy

* Fixes old code panic

* context, match, docs

* Addresses Shazniterinos, var names, expanded tests

* Returns subaccount name, provides USD values when offlinecalc

* Fixes oopsie

* Fixes cool bug which allowed made up subaccount results

* Subaccount override on FTX, subaccount results for collateral

* Strenghten collateral account info checks. Improve FTX test

* English is my first language

* Fixes oopsies

* Adds some conceptual futures order details to track PNL

* Initial design of future order processing in the backtester

* Introduces futures concept for collateral and spot/futures config diffs

* Fixes most tests

* Simple designs for collateral funding pair concept

* Expands interface use so much it hurts

* Implements more collateral interfaces

* Adds liquidation, adds strategy, struggles with Binance

* Attempts at getting FTX to work

* Adds calculatePNL as a wrapper function and adds an `IsFutures` asset check

* Successfully loads backtester with collateral currency

* Fails to really get much going for supporting futures

* Merges master changes

* Fleshes out how FTX processes collateral

* Further FTX collateral workings

* hooks up more ftx collateral and pnl calculations

* more funcs to flesh out handling

* Adds more links, just can't fit the pieces together :(

* Greatly expands futures order processing

* Fleshes out position tracker to also handle asset and exchange +testing

* RM linkedOrderID. rn positioncontroller, unexport

* Successfully tracks futures order positions

* Fails to calculate PNL

* Calculates pnl from orders accurately with exception to flipping orders

* Calculates PNL from orders

* Adds another controller layer to make it ez from orderstore

* Backtester now compiles. Adds test coverage

* labels things add scaling collateral test

* Calculates pnl in line with fees

* Mostly accurate PNL, with exception to appending with diff prices

* Adds locks, adds rpc function

* grpc implementations

* Gracefully handles rpc function

* beautiful tests!

* rejiggles tests to polish

* Finishes FTX testing, adds comments

* Exposes collateral calculations to rpc

* Adds commands and testing for rpcserver.go functions

* Increase testing and fix up backtester code

* Returns cool changes to original branch

* end of day fixes

* Fixing some tests

* Fixing tests 🎉

* Fixes all the tests

* Splits the backtester setup and running into different files

* Merge, minor fixes

* Messing with some strategy updates

* Failed understanding at collateral usage

* Begins the creation of cash and carry strategy

* Adds underlying pair, adds filldependentevent for futures

* Completes fill prerequsite event implementation. Can't short though

* Some bug fixes

* investigating funds

* CAN NOW CREATE A SHORT ORDER

* Minor change in short size

* Fixes for unrealised PNL & collateral rendering

* Fixes lint and tests

* Adds some verbosity

* Updates to pnl calc

* Tracks pnl for short orders, minor update to strategy

* Close and open event based on conditions

* Adds pnl data for currency statistics

* Working through PNL calculation automatically. Now panics

* Adds tracking, is blocked from design

* Work to flesh out closing a position

* vain attempts at tracking zeroing out bugs

* woww, super fun new subloggers 🎉

* Begins attempt at automatically handling contracts and collateral based on direction

* Merge master + fixes

* Investigating issues with pnl and holdings

* Minor pnl fixes

* Fixes future position sizing, needs contract sizing

* Can render pnl results, focussing on funding statistics

* tracking candles for futures, but why not btc

* Improves funding statistics

* Colours and stats

* Fixes collateral and snapshot bugs

* Completes test

* Fixes totals bug

* Fix double buy, expand stats, fixes usd totals, introduce interface

* Begins report formatting and calculations

* Appends pnl to receiving curr. Fixes map[time]. accurate USD

* Improves report output rendering

* PNL stats in report. New tests for futures

* Fixes existing tests before adding new coverage

* Test coverage

* Completes portfolio coverage

* Increase coverage exchange, portfolio. fix size bug. NEW CHART

* WHAT IS GOING ON WITH PNL

* Fixes PNL calculation. Adds ability to skip om futures tracking

* minor commit before merge

* Adds basic liquidation to backtester

* Changes liquidation to order based

* Liquidationnnnnn

* Further fleshes out liquidations

* Completes liquidations in a honorable manner. Adds AppendReasonf

* Beginnings of spot futures gap chart. Needs to link currencies to render difference

* Removes fake liquidation. Adds cool new chart

* Fixes somet tests,allows for zero fee value v nil distinction,New tests

* Some annoying test fixes that took too long

* portfolio coverage

* holding coverage, privatisation funding

* Testwork

* boring tests

* engine coverage

* More backtesting coverage

* Funding, strategy, report test coverage

* Completes coverage of report package

* Documentation, fixes some assumptions on asset errors

* Changes before master merge

* Lint and Tests

* defaults to non-coloured rendering

* Chart rendering

* Fixes surprise non-local-lints

* Niterinos to the extremeos

* Fixes merge problems

* The linter splintered across the glinting plinths

* Many nits addressed. Now sells spot position on final candle

* Adds forgotten coverage

* Adds ability to size futures contracts to match spot positions.

* fixes order sell sizing

* Adds tests to sizing. Fixes charting issue

* clint splintered the linters with flint

* Improves stats, stat rendering

* minifix

* Fixes tests and fee bug

* Merge fixeroos

* Microfixes

* Updates orderPNL on first Correctly utilises fees. Adds committed funds

* New base funcs. New order summary

* Fun test updates

* Fix logo colouring

* Fixes niteroonies

* Fix report

* BAD COMMIT

* Fixes funding issues.Updates default fee rates.Combines cashcarry case

* doc regen

* Now returns err

* Fixes sizing bug issue introduced in PR

* Fixes fun fee/total US value bug

* Fix chart bug. Show log charts with disclaimer

* sellside fee

* fixes fee and slippage view

* Fixed slippage price issue

* Fixes calculation and removes rendering

* Fixes stats and some rendering

* Merge fix

* Fixes merge issues

* go mod tidy, lint updates

* New linter attempt

* Version bump in appveyor and makefile

* Regex filename, config fixes, template h2 fixes

* Removes bad stats.

* neatens config builder. Moves filename generator

* Fixes issue where linter wants to fix my spelling

* Fixes pointers and starts
2022-06-30 15:43:41 +10:00

914 lines
22 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/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,
}
s.Reset()
if s.TotalOrders != 0 {
t.Error("expected 0")
}
}
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.SetupEventForTime(nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
err = s.SetupEventForTime(&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 err != nil {
t.Error(err)
}
if s.ExchangeAssetPairStatistics == nil {
t.Error("expected not nil")
}
if len(s.ExchangeAssetPairStatistics[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.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, errExchangeAssetPairStatsUnset) {
t.Errorf("received: %v, expected: %v", err, errExchangeAssetPairStatsUnset)
}
s.setupMap(exch, a)
s.ExchangeAssetPairStatistics = make(map[string]map[asset.Item]map[currency.Pair]*CurrencyPairStatistic)
b := &event.Base{}
err = s.SetEventForOffset(&signal.Signal{
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.SetupEventForTime(&kline.Kline{
Base: b,
Open: eleet,
Close: eleet,
Low: eleet,
High: eleet,
Volume: eleet,
})
if err != nil {
t.Error(err)
}
err = s.SetEventForOffset(&signal.Signal{
Base: b,
ClosePrice: eleet,
Direction: gctorder.Buy,
})
if err != nil {
t.Error(err)
}
}
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, errExchangeAssetPairStatsUnset) {
t.Errorf("received: %v, expected: %v", err, errExchangeAssetPairStatsUnset)
}
s.setupMap(exch, a)
s.ExchangeAssetPairStatistics = make(map[string]map[asset.Item]map[currency.Pair]*CurrencyPairStatistic)
b := &event.Base{}
err = s.SetEventForOffset(&order.Order{
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.SetupEventForTime(&kline.Kline{
Base: b,
Open: eleet,
Close: eleet,
Low: eleet,
High: eleet,
Volume: eleet,
})
if err != nil {
t.Error(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,
})
if err != nil {
t.Error(err)
}
}
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 err != nil && err.Error() != "exchangeAssetPairStatistics not setup" {
t.Error(err)
}
s.setupMap(exch, a)
s.ExchangeAssetPairStatistics = make(map[string]map[asset.Item]map[currency.Pair]*CurrencyPairStatistic)
b := &event.Base{}
err = s.SetEventForOffset(&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.SetupEventForTime(&kline.Kline{
Base: b,
Open: eleet,
Close: eleet,
Low: eleet,
High: eleet,
Volume: eleet,
})
if err != nil {
t.Error(err)
}
err = s.SetEventForOffset(&fill.Fill{
Base: b,
Direction: gctorder.Buy,
Amount: eleet,
ClosePrice: eleet,
VolumeAdjustedPrice: eleet,
PurchasePrice: eleet,
ExchangeFee: eleet,
Slippage: eleet,
})
if err != nil {
t.Error(err)
}
}
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.Pair]*CurrencyPairStatistic)
err = s.AddHoldingsForTime(&holdings.Holding{})
if !errors.Is(err, errCurrencyStatisticsUnset) {
t.Errorf("received: %v, expected: %v", err, errCurrencyStatisticsUnset)
}
err = s.SetupEventForTime(&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 err != nil {
t.Error(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,
})
if err != nil {
t.Error(err)
}
}
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(compliance.Snapshot{}, nil)
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.setupMap(exch, a)
s.ExchangeAssetPairStatistics = make(map[string]map[asset.Item]map[currency.Pair]*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.SetupEventForTime(&kline.Kline{
Base: b,
Open: eleet,
Close: eleet,
Low: eleet,
High: eleet,
Volume: eleet,
})
if err != nil {
t.Error(err)
}
err = s.AddComplianceSnapshotForTime(compliance.Snapshot{
Timestamp: tt,
}, &fill.Fill{
Base: b,
})
if err != nil {
t.Error(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.NewPair(currency.BTC, currency.USDT)
err := s.SetupEventForTime(nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
err = s.SetupEventForTime(&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 err != nil {
t.Error(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,
})
if err != nil {
t.Error(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,
})
if err != nil {
t.Error(err)
}
s.PrintAllEventsChronologically()
}
func TestCalculateTheResults(t *testing.T) {
t.Parallel()
s := Statistic{}
err := s.CalculateAllResults()
if !errors.Is(err, common.ErrNilArguments) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilArguments)
}
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.SetupEventForTime(nil)
if !errors.Is(err, common.ErrNilEvent) {
t.Errorf("received: %v, expected: %v", err, common.ErrNilEvent)
}
err = s.SetupEventForTime(&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 err != nil {
t.Error(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,
})
if err != nil {
t.Error(err)
}
err = s.SetupEventForTime(&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 err != nil {
t.Error(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,
})
if err != nil {
t.Error(err)
}
err = s.SetupEventForTime(&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 err != nil {
t.Error(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,
})
if err != nil {
t.Error(err)
}
err = s.SetupEventForTime(&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 err != nil {
t.Error(err)
}
err = s.SetEventForOffset(&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,
})
if err != nil {
t.Error(err)
}
s.ExchangeAssetPairStatistics[exch][a][p].Events[1].Holdings.QuoteInitialFunds = eleet
s.ExchangeAssetPairStatistics[exch][a][p].Events[1].Holdings.TotalValue = eleeet
s.ExchangeAssetPairStatistics[exch][a][p2].Events[1].Holdings.QuoteInitialFunds = eleet
s.ExchangeAssetPairStatistics[exch][a][p2].Events[1].Holdings.TotalValue = eleeet
funds, err := funding.SetupFundingManager(&engine.ExchangeManager{}, 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)
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, 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 []common.DataEventHandler
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 := []common.DataEventHandler{
&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)
}
}