Skip to content

Commit a7c716a

Browse files
committed
TST: Simplify test
1 parent 1740444 commit a7c716a

File tree

1 file changed

+3
-12
lines changed

1 file changed

+3
-12
lines changed

backtesting/test/_test.py

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -937,18 +937,9 @@ def test_FractionalBacktest(self):
937937
self.assertEqual(stats['# Trades'], 41)
938938
trades = stats['_trades']
939939
self.assertEqual(len(trades), 41)
940-
first_trade = trades[['Size', 'EntryPrice', 'ExitPrice', 'EntryBar']].head(1)
941-
self.assertEqual(first_trade['Size'][0], -0.422493) # Fractional value -422493
942-
self.assertAlmostEqual(first_trade['EntryPrice'][0], 236.69) # Fractional value 0.000236689
943-
self.assertAlmostEqual(first_trade['ExitPrice'][0], 261.7) # Fractional value 0.000261699
944-
indicators = stats['_strategy']._indicators
945-
self.assertEqual(len(indicators), 2)
946-
self.assertAlmostEqual(
947-
indicators[0][first_trade['EntryBar'][0]], 234.14, places=2
948-
) # Fractional value 0.000234139
949-
self.assertAlmostEqual(
950-
indicators[1][first_trade['EntryBar'][0]], 237.07, places=2
951-
) # Fractional value 0.000237067
940+
trade = trades.iloc[0]
941+
self.assertAlmostEqual(trade['EntryPrice'], 236.69)
942+
self.assertAlmostEqual(stats['_strategy']._indicators[0][trade['EntryBar']], 234.14)
952943

953944
def test_MultiBacktest(self):
954945
btm = MultiBacktest([GOOG, EURUSD, BTCUSD], SmaCross, cash=100_000)

0 commit comments

Comments
 (0)