From ec5318ec882e84fa240ef73217e7180adfae9d11 Mon Sep 17 00:00:00 2001 From: Richard Frank Date: Thu, 5 Nov 2015 10:15:01 -0500 Subject: [PATCH] TST: Fixed up test result comparison --- tests/test_algorithm.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_algorithm.py b/tests/test_algorithm.py index 5ebdf920..18e1c444 100644 --- a/tests/test_algorithm.py +++ b/tests/test_algorithm.py @@ -1907,7 +1907,7 @@ class TestTradingAlgorithm(TestCase): }) def test_analyze_called(self): - results_ref = None + self.perf_ref = None def initialize(context): pass @@ -1915,10 +1915,10 @@ class TestTradingAlgorithm(TestCase): def handle_data(context, data): pass - def analyze(self, perf): + def analyze(context, perf): self.perf_ref = perf algo = TradingAlgorithm(initialize=initialize, handle_data=handle_data, analyze=analyze) results = algo.run(self.panel) - self.assertEqual(results, results_ref) + self.assertIs(results, self.perf_ref)