Improved ATR code coverage

msarm · Apr 22, 2018 · 8f79eca · 8f79eca
1 parent 52625b3
commit 8f79eca
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/testcases/technical_atr_test.py b/testcases/technical_atr_test.py
@@ -25,6 +25,7 @@
 from pyalgotrade.technical import atr
 from pyalgotrade import bar
 from pyalgotrade.dataseries import bards
+from pyalgotrade import dataseries
 
 
 class TestCase(common.TestCase):
@@ -62,3 +63,8 @@ def testStockChartsATRAdjusted(self):
                 self.assertEqual(atrDS[-1], None)
             else:
                 self.assertEqual(common.safe_round(atrDS[-1], 2), round(expected[i]/2, 2))
+
+    def testInvalidDataSeries(self):
+        with self.assertRaisesRegexp(Exception, "barDataSeries must be a dataseries.bards.BarDataSeries instance"):
+            ds = dataseries.SequenceDataSeries()
+            atrDS = atr.ATR(ds, 14, True)