This commit is contained in:
Alexander Belopolsky 2012-04-29 18:20:05 -04:00
commit 934cb18ef5

View file

@ -1106,12 +1106,16 @@ def run_with_locale(catstr, *locales):
def run_with_tz(tz): def run_with_tz(tz):
def decorator(func): def decorator(func):
def inner(*args, **kwds): def inner(*args, **kwds):
try:
tzset = time.tzset
except AttributeError:
raise unittest.SkipTest("tzset required")
if 'TZ' in os.environ: if 'TZ' in os.environ:
orig_tz = os.environ['TZ'] orig_tz = os.environ['TZ']
else: else:
orig_tz = None orig_tz = None
os.environ['TZ'] = tz os.environ['TZ'] = tz
time.tzset() tzset()
# now run the function, resetting the tz on exceptions # now run the function, resetting the tz on exceptions
try: try: