nose2 - parametrised tests

Siddharta Govindaraj · Siddharta Govindaraj · commit ac686a1a23e3 · 2015-03-25T09:22:35.000+05:30
diff --git a/stock_alerter/tests/test_stock.py b/stock_alerter/tests/test_stock.py
@@ -2,6 +2,8 @@
 import collections
 from datetime import datetime, timedelta
 
+from nose2.tools.params import params
+
 from ..stock import Stock, StockSignal
 
 
@@ -21,6 +23,24 @@ def test_price_of_a_new_stock_class_should_be_None():
 test_price_of_a_new_stock_class_should_be_None.teardown = teardown_test
 
 
+def given_a_series_of_prices(stock, prices):
+    timestamps = [datetime(2014, 2, 10), datetime(2014, 2, 11),
+                  datetime(2014, 2, 12), datetime(2014, 2, 13)]
+    for timestamp, price in zip(timestamps, prices):
+        stock.update(timestamp, price)
+
+
+@params(
+    ([8, 10, 12], True),
+    ([8, 12, 10], False),
+    ([8, 10, 10], False)
+)
+def test_stock_trends(prices, expected_output):
+    goog = Stock("GOOG")
+    given_a_series_of_prices(goog, prices)
+    assert goog.is_increasing_trend() == expected_output
+
+
 class StockTest(unittest.TestCase):
     def setUp(self):
         self.goog = Stock("GOOG")
