From 50f4e1fff2f0e37c3c8dd75f814362bcf37b8206 Mon Sep 17 00:00:00 2001 From: Eddie Hebert Date: Fri, 28 Dec 2012 13:46:44 -0500 Subject: [PATCH] Fixes bug in test where period_end is too early. Grabs period end before we remove trades. --- tests/test_perf_tracking.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_perf_tracking.py b/tests/test_perf_tracking.py index 8efbc2be..a172fee4 100644 --- a/tests/test_perf_tracking.py +++ b/tests/test_perf_tracking.py @@ -625,6 +625,8 @@ class TestPerformanceTracker(unittest.TestCase): source_id="factory2" ) + period_end = trade_history2[-1].dt + # 'middle' start of 3 depends on number of days == 7 middle = 3 @@ -647,7 +649,7 @@ class TestPerformanceTracker(unittest.TestCase): trading_environment.period_start = \ trade_history[0].dt.replace(hour=0, minute=0, second=0) - trading_environment.period_end = trade_history2[-1].dt + trading_environment.period_end = period_end trading_environment.first_open = \ trading_environment.calculate_first_open() trading_environment.last_close = \