From ff485950b8305553dc592c639c2ea7e5af5a35f7 Mon Sep 17 00:00:00 2001 From: Richard Frank Date: Tue, 2 Feb 2016 10:06:17 -0500 Subject: [PATCH] TST: Fix expected value now that assertion is more strict in newer pandas --- tests/test_history.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_history.py b/tests/test_history.py index a3c9ae5b..9179d3c7 100644 --- a/tests/test_history.py +++ b/tests/test_history.py @@ -348,9 +348,9 @@ def handle_data(context, data): output = test_algo.run(source) self.assertIsNotNone(output) - history_trace = test_algo.history_trace + df.columns = self.env.asset_finder.retrieve_all(df.columns) - for i, received in enumerate(history_trace[bar_count - 1:]): + for i, received in enumerate(test_algo.history_trace[bar_count - 1:]): expected = df.iloc[i:i + bar_count] assert_frame_equal(expected, received)