Adapt analyzer tests to py3

mementum · mementum · commit 7b43300f7b30 · 2016-09-02T09:13:31.000+02:00
diff --git a/tests/test_analyzer-sqn.py b/tests/test_analyzer-sqn.py
@@ -27,6 +27,7 @@
 
 import backtrader as bt
 import backtrader.indicators as btind
+from backtrader.utils.py3 import PY2
 
 
 class TestStrategy(bt.Strategy):
@@ -157,7 +158,11 @@ def test_run(main=False):
             print(analysis)
             print(str(analysis.sqn))
         else:
-            assert str(analysis.sqn) == '0.912550316439'
+            # Handle different precision
+            if PY2:
+                assert str(analysis.sqn) == '0.912550316439'
+            else:
+                assert str(analysis.sqn) == '0.9125503164393756'
             assert str(analysis.trades) == '11'
 
 
diff --git a/tests/test_analyzer-timereturn.py b/tests/test_analyzer-timereturn.py
@@ -27,6 +27,7 @@
 
 import backtrader as bt
 import backtrader.indicators as btind
+from backtrader.utils.py3 import PY2
 
 
 class TestStrategy(bt.Strategy):
@@ -157,9 +158,15 @@ def test_run(main=False):
         analysis = analyzer.get_analysis()
         if main:
             print(analysis)
-            print(str(analysis[analysis.keys[0]]))
+            print(str(analysis[next(iter(analysis.keys()))]))
         else:
-            assert str(analysis[analysis.keys()[0]]) == '0.2795'
+            # Handle different precision
+            if PY2:
+                sval = '0.2795'
+            else:
+                sval = '0.2794999999999983'
+
+            assert str(analysis[next(iter(analysis.keys()))]) == sval
 
 
 if __name__ == '__main__':
