Add missing testsetup blocks for doctests
Fixes doctests that fail due to missing imports
diff --git a/docs/examples.rst b/docs/examples.rst
index 7d2181e..d706cd6 100644
--- a/docs/examples.rst
+++ b/docs/examples.rst
@@ -749,6 +749,19 @@
-----------------
Daily, for 7 days, jumping Saturday and Sunday occurrences.
+.. testsetup:: rruleset
+
+ import datetime
+
+ from dateutil.parser import parse
+ from dateutil.rrule import rrule, rruleset
+ from dateutil.rrule import YEARLY, MONTHLY, WEEKLY, DAILY
+ from dateutil.rrule import MO, TU, WE, TH, FR, SA, SU
+
+ import pprint
+ import sys
+ sys.displayhook = pprint.pprint
+
.. doctest:: rruleset
:options: +NORMALIZE_WHITESPACE
@@ -787,6 +800,15 @@
Every 10 days, 5 occurrences.
+.. testsetup:: rrulestr
+
+ from dateutil.parser import parse
+ from dateutil.rrule import rruleset, rrulestr
+
+ import pprint
+ import sys
+ sys.displayhook = pprint.pprint
+
.. doctest:: rrulestr
:options: +NORMALIZE_WHITESPACE
@@ -1252,6 +1274,12 @@
[http://www.python.org/doc/current/lib/module-time.html time module documentation].
+.. testsetup:: tzstr
+
+ import os
+ import time
+ from datetime import datetime
+ from dateutil.tz import tzstr
.. doctest:: tzstr
@@ -1304,6 +1332,10 @@
tzrange examples
----------------
+.. testsetup:: tzrange
+
+ from dateutil.tz import tzrange, tzstr
+
.. doctest:: tzrange
>>> tzstr('EST5EDT') == tzrange("EST", -18000, "EDT")
@@ -1334,6 +1366,11 @@
tzfile examples
---------------
+.. testsetup:: tzfile
+
+ from datetime import datetime
+ from dateutil.tz import tzfile, tzutc
+
.. doctest:: tzfile
:options: +NORMALIZE_WHITESPACE