Passing doctest directives via the test wrapper

Siddharta Govindaraj · Siddharta Govindaraj · commit b572b41d1200 · 2015-03-25T09:04:54.000+05:30
diff --git a/stock_alerter/readme.txt b/stock_alerter/readme.txt
@@ -8,7 +8,7 @@ are going to be processed. A simple dictionary will do.
 
 >>> from stock_alerter.stock import Stock
 >>> exchange = {"GOOG": Stock("GOOG"), "AAPL": Stock("AAPL")}
->>> for key in sorted(exchange.keys()): #doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
+>>> for key in sorted(exchange.keys()):
 ...    print(key, exchange[key])
 ...    
 AAPL       <stock_alerter.stock.Stock object at 0x0...>
diff --git a/stock_alerter/tests/test_doctest.py b/stock_alerter/tests/test_doctest.py
@@ -14,5 +14,6 @@ def load_tests(loader, tests, pattern):
         "datetime": datetime,
         "Stock": stock.Stock
     }, setUp=setup_stock_doctest))
-    tests.addTests(doctest.DocFileSuite("readme.txt", package="stock_alerter"))
+    options = doctest.ELLIPSIS | doctest.NORMALIZE_WHITESPACE
+    tests.addTests(doctest.DocFileSuite("readme.txt", package="stock_alerter", optionflags=options))
     return tests
