Files
gocryptotrader/backtester/report/report_test.go
Adrian Gallagher 0fd433e865 exchanges/bittrex,kucoin: Remove exchange implementation and fix minor test issue (#1403)
* exchanges/Bittrex: Remove exchange implementation

* Kucoin: Fix TestProcessMarketSnapshot after pair removal update

* Kucoin: Fix race due to duplicate setupWS call

Unleash your inner Max Verstappen

* Kucoin: Actually test spot/margin market snapshot replication
2023-11-22 10:20:09 +11:00

556 lines
15 KiB
Go

package report
import (
"errors"
"testing"
"time"
"github.com/shopspring/decimal"
"github.com/thrasher-corp/gocryptotrader/backtester/config"
"github.com/thrasher-corp/gocryptotrader/backtester/eventhandlers/portfolio/compliance"
"github.com/thrasher-corp/gocryptotrader/backtester/eventhandlers/statistics"
"github.com/thrasher-corp/gocryptotrader/backtester/funding"
"github.com/thrasher-corp/gocryptotrader/common/key"
"github.com/thrasher-corp/gocryptotrader/currency"
"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"
func TestGenerateReport(t *testing.T) {
t.Parallel()
e := testExchange
a := asset.Spot
p := currency.NewPair(currency.BTC, currency.USDT)
d := Data{
Config: &config.Config{
StrategySettings: config.StrategySettings{
DisableUSDTracking: true,
},
},
OutputPath: t.TempDir(),
TemplatePath: "tpl.gohtml",
OriginalCandles: []*gctkline.Item{
{
Candles: []gctkline.Candle{
{
Time: time.Now(),
Open: 1337,
High: 1337,
Low: 1337,
Close: 1337,
Volume: 1337,
ValidationIssues: "hello world!",
},
},
},
},
EnhancedCandles: []EnhancedKline{
{
Exchange: e,
Asset: a,
Pair: p,
Interval: gctkline.OneHour,
Watermark: "Binance - SPOT - BTC-USDT",
Candles: []DetailedCandle{
{
UnixMilli: time.Date(2020, 12, 12, 0, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1337,
High: 1339,
Low: 1336,
Close: 1338,
Volume: 3,
VolumeColour: "rgba(47, 194, 27, 0.8)",
MadeOrder: true,
OrderDirection: gctorder.Buy,
OrderAmount: decimal.NewFromInt(1337),
Shape: "arrowUp",
Text: "hi",
Position: "aboveBar",
Colour: "green",
PurchasePrice: 50,
},
{
UnixMilli: time.Date(2020, 12, 12, 1, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1332,
High: 1332,
Low: 1330,
Close: 1331,
Volume: 2,
MadeOrder: true,
OrderDirection: gctorder.Buy,
OrderAmount: decimal.NewFromInt(1337),
Shape: "arrowUp",
Text: "hi",
Position: "aboveBar",
Colour: "green",
PurchasePrice: 50,
VolumeColour: "rgba(252, 3, 3, 0.8)",
},
{
UnixMilli: time.Date(2020, 12, 12, 2, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1337,
High: 1339,
Low: 1336,
Close: 1338,
Volume: 3,
MadeOrder: true,
OrderDirection: gctorder.Buy,
OrderAmount: decimal.NewFromInt(1337),
Shape: "arrowUp",
Text: "hi",
Position: "aboveBar",
Colour: "green",
PurchasePrice: 50,
VolumeColour: "rgba(47, 194, 27, 0.8)",
},
{
UnixMilli: time.Date(2020, 12, 12, 3, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1337,
High: 1339,
Low: 1336,
Close: 1338,
Volume: 3,
MadeOrder: true,
OrderDirection: gctorder.Buy,
OrderAmount: decimal.NewFromInt(1337),
Shape: "arrowUp",
Text: "hi",
Position: "aboveBar",
Colour: "green",
PurchasePrice: 50,
VolumeColour: "rgba(252, 3, 3, 0.8)",
},
{
UnixMilli: time.Date(2020, 12, 12, 4, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1337,
High: 1339,
Low: 1336,
Close: 1338,
Volume: 3,
VolumeColour: "rgba(47, 194, 27, 0.8)",
},
},
},
{
Exchange: "Bitstamp",
Asset: a,
Pair: currency.NewPair(currency.BTC, currency.USD),
Interval: gctkline.OneDay,
Watermark: "BITSTAMP - SPOT - BTC-USD - 1d",
Candles: []DetailedCandle{
{
UnixMilli: time.Date(2020, 12, 12, 0, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1337,
High: 1339,
Low: 1336,
Close: 1338,
Volume: 3,
MadeOrder: true,
OrderDirection: gctorder.Buy,
OrderAmount: decimal.NewFromInt(1337),
Shape: "arrowUp",
Text: "hi",
Position: "aboveBar",
Colour: "green",
PurchasePrice: 50,
VolumeColour: "rgba(47, 194, 27, 0.8)",
},
{
UnixMilli: time.Date(2020, 12, 12, 1, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1332,
High: 1332,
Low: 1330,
Close: 1331,
Volume: 2,
MadeOrder: true,
OrderDirection: gctorder.Buy,
OrderAmount: decimal.NewFromInt(1337),
Shape: "arrowUp",
Text: "hi",
Position: "aboveBar",
Colour: "green",
PurchasePrice: 50,
VolumeColour: "rgba(252, 3, 3, 0.8)",
},
{
UnixMilli: time.Date(2020, 12, 12, 2, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1337,
High: 1339,
Low: 1336,
Close: 1338,
Volume: 3,
MadeOrder: true,
OrderDirection: gctorder.Buy,
OrderAmount: decimal.NewFromInt(1337),
Shape: "arrowUp",
Text: "hi",
Position: "aboveBar",
Colour: "green",
PurchasePrice: 50,
VolumeColour: "rgba(47, 194, 27, 0.8)",
},
{
UnixMilli: time.Date(2020, 12, 12, 3, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1337,
High: 1339,
Low: 1336,
Close: 1338,
Volume: 3,
MadeOrder: true,
OrderDirection: gctorder.Buy,
OrderAmount: decimal.NewFromInt(1337),
Shape: "arrowUp",
Text: "hi",
Position: "aboveBar",
Colour: "green",
PurchasePrice: 50,
VolumeColour: "rgba(252, 3, 3, 0.8)",
},
{
UnixMilli: time.Date(2020, 12, 12, 4, 0, 0, 0, time.UTC).UnixMilli(),
Open: 1337,
High: 1339,
Low: 1336,
Close: 1338,
Volume: 3,
VolumeColour: "rgba(47, 194, 27, 0.8)",
},
},
},
},
Statistics: &statistics.Statistic{
FundingStatistics: &statistics.FundingStatistics{
Report: &funding.Report{
DisableUSDTracking: true,
},
Items: []statistics.FundingItemStatistics{
{
ReportItem: &funding.ReportItem{Snapshots: []funding.ItemSnapshot{{Time: time.Now()}}},
},
},
TotalUSDStatistics: &statistics.TotalFundingStatistics{},
},
StrategyName: "testStrat",
RiskFreeRate: decimal.NewFromFloat(0.03),
ExchangeAssetPairStatistics: map[key.ExchangePairAsset]*statistics.CurrencyPairStatistic{
{
Base: p.Base.Item,
Quote: p.Quote.Item,
Asset: a,
Exchange: e,
}: {
LowestClosePrice: statistics.ValueAtTime{Value: decimal.NewFromInt(100)},
HighestClosePrice: statistics.ValueAtTime{Value: decimal.NewFromInt(200)},
MarketMovement: decimal.NewFromInt(100),
StrategyMovement: decimal.NewFromInt(100),
CompoundAnnualGrowthRate: decimal.NewFromInt(1),
BuyOrders: 1,
SellOrders: 1,
ArithmeticRatios: &statistics.Ratios{},
GeometricRatios: &statistics.Ratios{},
},
},
TotalBuyOrders: 1337,
TotalSellOrders: 1330,
TotalOrders: 200,
BiggestDrawdown: &statistics.FinalResultsHolder{
Exchange: e,
Asset: a,
Pair: p,
MaxDrawdown: statistics.Swing{
Highest: statistics.ValueAtTime{
Time: time.Now(),
Value: decimal.NewFromInt(1337),
},
Lowest: statistics.ValueAtTime{
Time: time.Now(),
Value: decimal.NewFromInt(137),
},
DrawdownPercent: decimal.NewFromInt(100),
},
MarketMovement: decimal.NewFromInt(1377),
StrategyMovement: decimal.NewFromInt(1377),
},
BestStrategyResults: &statistics.FinalResultsHolder{
Exchange: e,
Asset: a,
Pair: p,
MaxDrawdown: statistics.Swing{
Highest: statistics.ValueAtTime{
Time: time.Now(),
Value: decimal.NewFromInt(1337),
},
Lowest: statistics.ValueAtTime{
Time: time.Now(),
Value: decimal.NewFromInt(137),
},
DrawdownPercent: decimal.NewFromInt(100),
},
MarketMovement: decimal.NewFromInt(1337),
StrategyMovement: decimal.NewFromInt(1337),
},
BestMarketMovement: &statistics.FinalResultsHolder{
Exchange: e,
Asset: a,
Pair: p,
MaxDrawdown: statistics.Swing{
Highest: statistics.ValueAtTime{
Time: time.Now(),
Value: decimal.NewFromInt(1337),
},
Lowest: statistics.ValueAtTime{
Time: time.Now(),
Value: decimal.NewFromInt(137),
},
DrawdownPercent: decimal.NewFromInt(100),
},
MarketMovement: decimal.NewFromInt(1337),
StrategyMovement: decimal.NewFromInt(1337),
},
},
}
if err := d.GenerateReport(); err != nil {
t.Error(err)
}
}
func TestEnhanceCandles(t *testing.T) {
t.Parallel()
tt := time.Now()
var d Data
err := d.enhanceCandles()
if !errors.Is(err, errNoCandles) {
t.Errorf("received: %v, expected: %v", err, errNoCandles)
}
err = d.SetKlineData(&gctkline.Item{})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = d.enhanceCandles()
if !errors.Is(err, errStatisticsUnset) {
t.Errorf("received: %v, expected: %v", err, errStatisticsUnset)
}
d.Statistics = &statistics.Statistic{}
err = d.enhanceCandles()
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
d.Statistics.ExchangeAssetPairStatistics = make(map[key.ExchangePairAsset]*statistics.CurrencyPairStatistic)
d.Statistics.ExchangeAssetPairStatistics[key.ExchangePairAsset{
Exchange: testExchange,
Base: currency.BTC.Item,
Quote: currency.USDT.Item,
Asset: asset.Spot,
}] = &statistics.CurrencyPairStatistic{}
err = d.SetKlineData(&gctkline.Item{
Exchange: testExchange,
Pair: currency.NewPair(currency.BTC, currency.USDT),
Asset: asset.Spot,
Interval: gctkline.OneDay,
Candles: []gctkline.Candle{
{
Time: tt,
Open: 1336,
High: 1338,
Low: 1336,
Close: 1337,
Volume: 1337,
},
},
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = d.enhanceCandles()
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = d.SetKlineData(&gctkline.Item{
Exchange: testExchange,
Pair: currency.NewPair(currency.BTC, currency.USDT),
Asset: asset.Spot,
Interval: gctkline.OneDay,
Candles: []gctkline.Candle{
{
Time: tt,
Open: 1336,
High: 1338,
Low: 1336,
Close: 1336,
Volume: 1337,
},
{
Time: tt,
Open: 1336,
High: 1338,
Low: 1336,
Close: 1335,
Volume: 1337,
},
},
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
err = d.enhanceCandles()
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
d.Statistics.ExchangeAssetPairStatistics[key.ExchangePairAsset{
Exchange: testExchange,
Base: currency.BTC.Item,
Quote: currency.USDT.Item,
Asset: asset.Spot,
}].FinalOrders = compliance.Snapshot{
Orders: []compliance.SnapshotOrder{
{
ClosePrice: decimal.NewFromInt(1335),
VolumeAdjustedPrice: decimal.NewFromInt(1337),
SlippageRate: decimal.NewFromInt(1),
CostBasis: decimal.NewFromInt(1337),
Order: nil,
},
},
Timestamp: tt,
}
err = d.enhanceCandles()
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
d.Statistics.ExchangeAssetPairStatistics[key.ExchangePairAsset{
Exchange: testExchange,
Base: currency.BTC.Item,
Quote: currency.USDT.Item,
Asset: asset.Spot,
}].FinalOrders = compliance.Snapshot{
Orders: []compliance.SnapshotOrder{
{
ClosePrice: decimal.NewFromInt(1335),
VolumeAdjustedPrice: decimal.NewFromInt(1337),
SlippageRate: decimal.NewFromInt(1),
CostBasis: decimal.NewFromInt(1337),
Order: &gctorder.Detail{
Date: tt,
Side: gctorder.Buy,
},
},
},
Timestamp: tt,
}
err = d.enhanceCandles()
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
d.Statistics.ExchangeAssetPairStatistics[key.ExchangePairAsset{
Exchange: testExchange,
Base: currency.BTC.Item,
Quote: currency.USDT.Item,
Asset: asset.Spot,
}].FinalOrders = compliance.Snapshot{
Orders: []compliance.SnapshotOrder{
{
ClosePrice: decimal.NewFromInt(1335),
VolumeAdjustedPrice: decimal.NewFromInt(1337),
SlippageRate: decimal.NewFromInt(1),
CostBasis: decimal.NewFromInt(1337),
Order: &gctorder.Detail{
Date: tt,
Side: gctorder.Sell,
},
},
},
Timestamp: tt,
}
err = d.enhanceCandles()
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
if len(d.EnhancedCandles) == 0 {
t.Error("expected enhanced candles")
}
}
func TestUpdateItem(t *testing.T) {
t.Parallel()
d := Data{}
tt := time.Now()
err := d.SetKlineData(&gctkline.Item{
Candles: []gctkline.Candle{
{
Time: tt,
},
},
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
if len(d.OriginalCandles) != 1 {
t.Fatal("expected Original Candles len of 1")
}
if len(d.OriginalCandles[0].Candles) != 1 {
t.Error("expected one candle")
}
err = d.SetKlineData(&gctkline.Item{
Candles: []gctkline.Candle{
{
Time: tt,
},
},
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
if len(d.OriginalCandles[0].Candles) != 1 {
t.Error("expected one candle")
}
err = d.SetKlineData(&gctkline.Item{
Candles: []gctkline.Candle{
{
Time: tt.Add(gctkline.OneMin.Duration()),
},
},
})
if !errors.Is(err, nil) {
t.Errorf("received: %v, expected: %v", err, nil)
}
if len(d.OriginalCandles[0].Candles) != 2 {
t.Error("expected two candles")
}
}
func TestCopyCloseFromPreviousEvent(t *testing.T) {
t.Parallel()
d := DetailedCandle{}
d.copyCloseFromPreviousEvent(&EnhancedKline{
Candles: []DetailedCandle{
{
Close: 1337,
},
},
})
if d.Close != 1337 {
t.Error("expected 1337")
}
}
func TestUseDarkMode(t *testing.T) {
t.Parallel()
d := Data{}
d.UseDarkMode(true)
if !d.UseDarkTheme {
t.Error("expected true")
}
}