mirror of
https://github.com/weewx/weewx.git
synced 2026-04-19 17:16:56 -04:00
503 lines
27 KiB
Python
503 lines
27 KiB
Python
#
|
|
# Copyright (c) 2009, 2010, 2011 Tom Keffer <tkeffer@gmail.com>
|
|
#
|
|
# See the file LICENSE.txt for your full rights.
|
|
#
|
|
# $Revision$
|
|
# $Author$
|
|
# $Date$
|
|
#
|
|
"""Test routines for weeutil.weeutil."""
|
|
|
|
import unittest
|
|
import calendar
|
|
import os
|
|
import time
|
|
|
|
from weeutil.weeutil import startOfInterval, option_as_list, TimeSpan, genYearSpans, genMonthSpans, genDaySpans
|
|
from weeutil.weeutil import archiveDaySpan, archiveWeekSpan, archiveMonthSpan, archiveYearSpan, archiveRainYearSpan
|
|
from weeutil.weeutil import startOfDay, startOfArchiveDay, intervalgen, stampgen
|
|
|
|
class WeeutilTest(unittest.TestCase):
|
|
|
|
def test_option_as_list(self):
|
|
|
|
self.assertEqual(option_as_list("abc"), ['abc'])
|
|
self.assertEqual(option_as_list(['a', 'b']), ['a', 'b'])
|
|
self.assertEqual(option_as_list(None), None)
|
|
self.assertEqual(option_as_list(''), [''])
|
|
|
|
def test_stampgen(self):
|
|
|
|
# Test the start of DST using a 30 minute increment:
|
|
start = time.mktime((2013,3,10,0,0,0,0,0,-1))
|
|
stop = time.mktime((2013,3,10,6,0,0,0,0,-1))
|
|
for ts, check_ts in zip(stampgen(start, stop, 1800), [1362902400,1362904200,1362906000,1362907800,
|
|
1362909600,1362911400,1362913200,1362915000,
|
|
1362916800,1362918600,1362920400]):
|
|
self.assertEqual(ts, check_ts)
|
|
|
|
# Test the ending of DST using a 30 minute increment:
|
|
start = time.mktime((2013,11,3, 0,0,0,0,0,-1))
|
|
stop = time.mktime((2013,11,3, 6,0,0,0,0,-1))
|
|
for ts, check_ts in zip(stampgen(start, stop, 1800), [1383462000,1383463800,1383465600,1383467400,
|
|
1383472800,1383474600,1383476400,1383478200,
|
|
1383480000,1383481800,1383483600,1383485400,
|
|
1383487200]):
|
|
self.assertEqual(ts, check_ts)
|
|
|
|
# Test the start of DST using a 3 hour increment
|
|
start = time.mktime((2013,3, 9,12,0,0,0,0,-1))
|
|
stop = time.mktime((2013,3,10,11,0,0,0,0,-1))
|
|
for ts, check_ts in zip(stampgen(start, stop,10800), [1362859200,1362870000,1362880800,1362891600,
|
|
1362902400,1362909600,1362920400,1362931200]):
|
|
self.assertEqual(ts, check_ts)
|
|
|
|
# Test the end of DST using a 3 hour increment
|
|
start = time.mktime((2013,11,2,12,0,0,0,0,-1))
|
|
stop = time.mktime((2013,11,3,12,0,0,0,0,-1))
|
|
for ts, check_ts in zip(stampgen(start, stop,10800), [1383418800,1383429600,1383440400,1383451200,
|
|
1383462000,1383476400,1383487200,1383498000,
|
|
1383508800]):
|
|
self.assertEqual(ts, check_ts)
|
|
|
|
def test_intervalgen(self):
|
|
|
|
# Test the start of DST using a 30 minute increment:
|
|
start = time.mktime((2013,3,10, 0,0,0,0,0,-1))
|
|
stop = time.mktime((2013,3,10, 5,0,0,0,0,-1))
|
|
for s, check_s in zip(intervalgen(start, stop, 1800), [(1362902400, 1362904200),(1362904200, 1362906000),
|
|
(1362906000, 1362907800),(1362907800, 1362909600),
|
|
(1362909600, 1362911400),(1362911400, 1362913200),
|
|
(1362913200, 1362915000),(1362915000, 1362916800)]):
|
|
self.assertEqual(s, TimeSpan(check_s[0], check_s[1]))
|
|
|
|
# Test the ending of DST using a 30 minute increment:
|
|
start = time.mktime((2013,11,3, 0,0,0,0,0,-1))
|
|
stop = time.mktime((2013,11,3, 6,0,0,0,0,-1))
|
|
for s, check_s in zip(intervalgen(start, stop, 1800), [(1383462000, 1383463800),(1383463800, 1383465600),
|
|
(1383465600, 1383467400),(1383467400, 1383472800),
|
|
(1383472800, 1383474600),(1383474600, 1383476400),
|
|
(1383476400, 1383478200),(1383478200, 1383480000),
|
|
(1383480000, 1383481800),(1383481800, 1383483600),
|
|
(1383483600, 1383485400),(1383485400, 1383487200)]):
|
|
self.assertEqual(s, TimeSpan(check_s[0], check_s[1]))
|
|
|
|
# Test the start of DST using a 3 hour increment:
|
|
start = time.mktime((2013,3, 9,12,0,0,0,0,-1))
|
|
stop = time.mktime((2013,3,10,11,0,0,0,0,-1))
|
|
for s, check_s in zip(intervalgen(start, stop,10800), [(1362859200, 1362870000),(1362870000, 1362880800),
|
|
(1362880800, 1362891600),(1362891600, 1362902400),
|
|
(1362902400, 1362909600),(1362909600, 1362920400),
|
|
(1362920400, 1362931200),(1362931200, 1362938400)]):
|
|
self.assertEqual(s, TimeSpan(check_s[0], check_s[1]))
|
|
|
|
# Test the ending of DST using a 3 hour increment:
|
|
start = time.mktime((2013,11,2,12,0,0,0,0,-1))
|
|
stop = time.mktime((2013,11,3,12,0,0,0,0,-1))
|
|
for s, check_s in zip(intervalgen(start, stop,10800), [(1383418800, 1383429600),(1383429600, 1383440400),
|
|
(1383440400, 1383451200),(1383451200, 1383462000),
|
|
(1383462000, 1383476400),(1383476400, 1383487200),
|
|
(1383487200, 1383498000),(1383498000, 1383508800)]):
|
|
self.assertEqual(s, TimeSpan(check_s[0], check_s[1]))
|
|
|
|
|
|
def test_startOfInterval(self):
|
|
|
|
t_length = 1 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 57, 17, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 1, 57, 0, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
t_length = 5 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 57, 17, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 1, 55, 0, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
t_length = 1 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 0, 0, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 0, 59, 0, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
t_length = 5 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 0, 0, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 0, 55, 0, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
t_length = 10 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 57, 17, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 1, 50, 0, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
t_length = 15 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 57, 17, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 1, 45, 0, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
t_length = 20 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 57, 17, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 1, 40, 0, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
t_length = 30 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 57, 17, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 1, 30, 00, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
t_length = 60 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 57, 17, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 1, 00, 00, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
t_length = 120 * 60
|
|
t_test = time.mktime((2009, 3, 4, 1, 57, 17, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 3, 4, 0, 00, 00, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
# Do a test over the spring DST boundary
|
|
# This is 03:22:05 DST, just after the change over.
|
|
# The correct answer is 02:00:00 DST.
|
|
t_length = 120 * 60
|
|
t_test = time.mktime((2009, 3, 8, 3, 22, 05, 0, 0, 1))
|
|
t_ans = time.mktime((2009, 3, 8, 2, 00, 00, 0, 0, 1))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
# Do a test over the fall DST boundary.
|
|
# This is 01:22:05 DST, just before the change over.
|
|
# The correct answer is 00:00:00 DST.
|
|
t_length = 120 * 60
|
|
t_test = time.mktime((2009, 11, 1, 1, 22, 05, 0, 0, 1))
|
|
t_ans = time.mktime((2009, 11, 1, 0, 0, 0, 0, 0, 1))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
# Do it again, except after the change over
|
|
# This is 01:22:05 ST, just after the change over.
|
|
# The correct answer is 00:00:00 ST (which is 01:00:00 DST).
|
|
t_length = 120 * 60
|
|
t_test = time.mktime((2009, 11, 1, 1, 22, 05, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 11, 1, 0, 00, 00, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
# Once again at 01:22:05 ST, just before the change over, but w/shorter interval
|
|
t_length = 5 * 60
|
|
t_test = time.mktime((2009, 11, 1, 1, 22, 05, 0, 0, 1))
|
|
t_ans = time.mktime((2009, 11, 1, 1, 20, 00, 0, 0, 1))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
# Once again at 01:22:05 ST, just after the change over, but w/shorter interval
|
|
t_length = 5 * 60
|
|
t_test = time.mktime((2009, 11, 1, 1, 22, 05, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 11, 1, 1, 20, 00, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
# Once again at 01:22:05 ST, just after the change over, but with 1 hour interval
|
|
t_length = 60 * 60
|
|
t_test = time.mktime((2009, 11, 1, 1, 22, 05, 0, 0, 0))
|
|
t_ans = time.mktime((2009, 11, 1, 1, 00, 00, 0, 0, 0))
|
|
t_start = startOfInterval(t_test, t_length)
|
|
self.assertEqual(t_start, t_ans)
|
|
|
|
def test_TimeSpans(self):
|
|
|
|
t = TimeSpan(1230000000, 1231000000)
|
|
# Reflexive test:
|
|
self.assertEqual(t, t)
|
|
tsub = TimeSpan(1230500000, 1230600000)
|
|
self.assertTrue(t.includes(tsub))
|
|
self.assertFalse(tsub.includes(t))
|
|
tleft = TimeSpan(1229000000, 1229100000)
|
|
self.assertFalse(t.includes(tleft))
|
|
tright = TimeSpan(1232000000, 1233000000)
|
|
self.assertFalse(t.includes(tright))
|
|
|
|
dic={}
|
|
dic[t] = 't'
|
|
dic[tsub] = 'tsub'
|
|
dic[tleft] = 'tleft'
|
|
dic[tright] = 'tright'
|
|
|
|
self.assertEqual(dic[t], 't')
|
|
|
|
def test_genYearSpans(self):
|
|
|
|
# Should generate years 2007 through 2008:"
|
|
start_ts = time.mktime((2007, 12, 3, 10, 15, 0, 0, 0, -1))
|
|
stop_ts = time.mktime((2008, 3, 1, 0, 0, 0, 0, 0, -1))
|
|
|
|
yearlist = [span for span in genYearSpans(start_ts, stop_ts)]
|
|
|
|
expected = ["[2007-01-01 00:00:00 PST (1167638400) -> 2008-01-01 00:00:00 PST (1199174400)]",
|
|
"[2008-01-01 00:00:00 PST (1199174400) -> 2009-01-01 00:00:00 PST (1230796800)]"]
|
|
|
|
for got, expect in zip(yearlist, expected):
|
|
self.assertEqual(str(got), expect)
|
|
|
|
def test_genMonthSpans(self):
|
|
# Should generate months 2007-12 through 2008-02:
|
|
start_ts = time.mktime((2007, 12, 3, 10, 15, 0, 0, 0, -1))
|
|
stop_ts = time.mktime((2008, 3, 1, 0, 0, 0, 0, 0, -1))
|
|
|
|
monthlist = [span for span in genMonthSpans(start_ts, stop_ts)]
|
|
|
|
expected = ["[2007-12-01 00:00:00 PST (1196496000) -> 2008-01-01 00:00:00 PST (1199174400)]",
|
|
"[2008-01-01 00:00:00 PST (1199174400) -> 2008-02-01 00:00:00 PST (1201852800)]",
|
|
"[2008-02-01 00:00:00 PST (1201852800) -> 2008-03-01 00:00:00 PST (1204358400)]"]
|
|
|
|
for got, expect in zip(monthlist, expected):
|
|
self.assertEqual(str(got), expect)
|
|
|
|
# Add a second to the stop time. This should generate months 2007-12 through 2008-03:"
|
|
start_ts = time.mktime((2007, 12, 3, 10, 15, 0, 0, 0, -1))
|
|
stop_ts = time.mktime((2008, 3, 1, 0, 0, 1, 0, 0, -1))
|
|
|
|
monthlist = [span for span in genMonthSpans(start_ts, stop_ts)]
|
|
|
|
expected = ["[2007-12-01 00:00:00 PST (1196496000) -> 2008-01-01 00:00:00 PST (1199174400)]",
|
|
"[2008-01-01 00:00:00 PST (1199174400) -> 2008-02-01 00:00:00 PST (1201852800)]",
|
|
"[2008-02-01 00:00:00 PST (1201852800) -> 2008-03-01 00:00:00 PST (1204358400)]",
|
|
"[2008-03-01 00:00:00 PST (1204358400) -> 2008-04-01 00:00:00 PDT (1207033200)]"]
|
|
|
|
for got, expect in zip(monthlist, expected):
|
|
self.assertEqual(str(got), expect)
|
|
|
|
def test_genDaySpans(self):
|
|
|
|
# Should generate 2007-12-23 through 2008-1-5:"
|
|
start_ts = time.mktime((2007, 12, 23, 10, 15, 0, 0, 0, -1))
|
|
stop_ts = time.mktime((2008, 1, 5, 9, 22, 0, 0, 0, -1))
|
|
|
|
daylist = [span for span in genDaySpans(start_ts, stop_ts)]
|
|
|
|
expected = ["[2007-12-23 00:00:00 PST (1198396800) -> 2007-12-24 00:00:00 PST (1198483200)]",
|
|
"[2007-12-24 00:00:00 PST (1198483200) -> 2007-12-25 00:00:00 PST (1198569600)]",
|
|
"[2007-12-25 00:00:00 PST (1198569600) -> 2007-12-26 00:00:00 PST (1198656000)]",
|
|
"[2007-12-26 00:00:00 PST (1198656000) -> 2007-12-27 00:00:00 PST (1198742400)]",
|
|
"[2007-12-27 00:00:00 PST (1198742400) -> 2007-12-28 00:00:00 PST (1198828800)]",
|
|
"[2007-12-28 00:00:00 PST (1198828800) -> 2007-12-29 00:00:00 PST (1198915200)]",
|
|
"[2007-12-29 00:00:00 PST (1198915200) -> 2007-12-30 00:00:00 PST (1199001600)]",
|
|
"[2007-12-30 00:00:00 PST (1199001600) -> 2007-12-31 00:00:00 PST (1199088000)]",
|
|
"[2007-12-31 00:00:00 PST (1199088000) -> 2008-01-01 00:00:00 PST (1199174400)]",
|
|
"[2008-01-01 00:00:00 PST (1199174400) -> 2008-01-02 00:00:00 PST (1199260800)]",
|
|
"[2008-01-02 00:00:00 PST (1199260800) -> 2008-01-03 00:00:00 PST (1199347200)]",
|
|
"[2008-01-03 00:00:00 PST (1199347200) -> 2008-01-04 00:00:00 PST (1199433600)]",
|
|
"[2008-01-04 00:00:00 PST (1199433600) -> 2008-01-05 00:00:00 PST (1199520000)]",
|
|
"[2008-01-05 00:00:00 PST (1199520000) -> 2008-01-06 00:00:00 PST (1199606400)]"]
|
|
|
|
for got, expect in zip(daylist, expected):
|
|
self.assertEqual(str(got), expect)
|
|
|
|
# Should generate the single date 2007-12-1:"
|
|
daylist = [span for span in genDaySpans(time.mktime((2007, 12, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 12, 2, 0, 0, 0, 0, 0, -1)))]
|
|
|
|
expected = ["[2007-12-01 00:00:00 PST (1196496000) -> 2007-12-02 00:00:00 PST (1196582400)]"]
|
|
for got, expect in zip(daylist, expected):
|
|
self.assertEqual(str(got), expect)
|
|
|
|
def test_daySpan(self):
|
|
self.assertEqual(archiveDaySpan(time.mktime((2007, 12, 13, 10, 15, 0, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 12, 13, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 12, 14, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveDaySpan(time.mktime((2007, 12, 13, 0, 0, 0, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 12, 12, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 12, 13, 0, 0, 0, 0, 0, -1))))
|
|
# Try it again with grace=0
|
|
self.assertEqual(archiveDaySpan(time.mktime((2007, 12, 13, 0, 0, 0, 0, 0, -1)), grace=0),
|
|
TimeSpan(time.mktime((2007, 12, 13, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 12, 14, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveDaySpan(time.mktime((2007, 12, 13, 0, 0, 1, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 12, 13, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 12, 14, 0, 0, 0, 0, 0, -1))))
|
|
|
|
def test_weekSpan(self):
|
|
|
|
self.assertEqual(archiveWeekSpan(time.mktime((2007, 12, 13, 10, 15, 0, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 12, 9, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 12, 16, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveWeekSpan(time.mktime((2007, 12, 9, 0, 0, 0, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 12, 2, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 12, 9, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveWeekSpan(time.mktime((2007, 12, 9, 0, 0, 1, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 12, 9, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 12, 16, 0, 0, 0, 0, 0, -1))))
|
|
|
|
def test_monthSpan(self):
|
|
|
|
self.assertEqual(archiveMonthSpan(time.mktime((2007, 12, 13, 10, 15, 0, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 12, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2008, 1, 1, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveMonthSpan(time.mktime((2007, 12, 1, 0, 0, 0, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 11, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 12, 1, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveMonthSpan(time.mktime((2007, 12, 1, 0, 0, 1, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 12, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2008, 1, 1, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveMonthSpan(time.mktime((2008, 1, 1, 0, 0, 0, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 12, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2008, 1, 1, 0, 0, 0, 0, 0, -1))))
|
|
|
|
def test_yearSpan(self):
|
|
|
|
self.assertEqual(archiveYearSpan(time.mktime((2007, 12, 13, 10, 15, 0, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 1, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2008, 1, 1, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveYearSpan(time.mktime((2008, 1, 1, 0, 0, 0, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2007, 1, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2008, 1, 1, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveYearSpan(time.mktime((2008, 1, 1, 0, 0, 1, 0, 0, -1))),
|
|
TimeSpan(time.mktime((2008, 1, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2009, 1, 1, 0, 0, 0, 0, 0, -1))))
|
|
|
|
def test_rainYearSpan(self):
|
|
|
|
self.assertEqual(archiveRainYearSpan(time.mktime((2007, 2, 13, 10, 15, 0, 0, 0, -1)), 10),
|
|
TimeSpan(time.mktime((2006, 10, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2007, 10, 1, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(archiveRainYearSpan(time.mktime((2007, 12, 13, 10, 15, 0, 0, 0, -1)), 10),
|
|
TimeSpan(time.mktime((2007, 10, 1, 0, 0, 0, 0, 0, -1)),
|
|
time.mktime((2008, 10, 1, 0, 0, 0, 0, 0, -1))))
|
|
|
|
def test_DST(self):
|
|
|
|
# Test start-of-day routines around a DST boundary:
|
|
start_ts = time.mktime((2007, 3, 11, 1, 0, 0, 0, 0, -1))
|
|
start_of_day = startOfDay(start_ts)
|
|
start2 = startOfArchiveDay(start_of_day)
|
|
|
|
# Check that this is, in fact, a DST boundary:
|
|
self.assertEqual(start_of_day, int(time.mktime((2007, 3, 11, 0, 0, 0, 0, 0, -1))))
|
|
self.assertEqual(start2 , int(time.mktime((2007, 3, 10, 0, 0, 0, 0, 0, -1))))
|
|
|
|
def test_dnt(self):
|
|
"""test day/night transitions"""
|
|
from weeutil.weeutil import timestamp_to_gmtime
|
|
from weeutil.weeutil import timestamp_to_string
|
|
from weeutil.weeutil import getDayNightTransitions
|
|
tz = os.environ['TZ']
|
|
times = [(calendar.timegm((2012, 1, 2, 0, 0, 0, 0, 0, -1)),
|
|
calendar.timegm((2012, 1, 3, 0, 0, 0, 0, 0, -1))),
|
|
(calendar.timegm((2012, 1, 2, 22, 0, 0, 0, 0, -1)),
|
|
calendar.timegm((2012, 1, 3, 22, 0, 0, 0, 0, -1)))]
|
|
locs = [(-33.86, 151.21, 'sydney', 'Australia/Sydney'), # UTC+10:00
|
|
(35.6895, 139.6917, 'tokyo', 'Asia/Tokyo'), # UTC+09:00
|
|
(-33.93, 18.42, 'cape town', 'Africa/Johannesburg'), # UTC+02:00
|
|
(51.4791, 0, 'greenwich', 'Europe/London'), # UTC 00:00
|
|
(42.358, -71.060, 'boston', 'America/New_York'), # UTC-05:00
|
|
(21.3, -157.8167, 'honolulu', 'Pacific/Honolulu'), # UTC-10:00
|
|
]
|
|
expected = [
|
|
(
|
|
("lat: -33.86 lon: 151.21 sydney first: day",
|
|
"2012-01-02 00:00:00 UTC (1325462400) 2012-01-02 11:00:00 EST (1325462400)",
|
|
"2012-01-02 09:09:33 UTC (1325495373) 2012-01-02 20:09:33 EST (1325495373)",
|
|
"2012-01-02 18:48:44 UTC (1325530124) 2012-01-03 05:48:44 EST (1325530124)",
|
|
"2012-01-03 00:00:00 UTC (1325548800) 2012-01-03 11:00:00 EST (1325548800)",
|
|
),
|
|
("lat: 35.6895 lon: 139.6917 tokyo first: day",
|
|
"2012-01-02 00:00:00 UTC (1325462400) 2012-01-02 09:00:00 CJT (1325462400)",
|
|
"2012-01-02 07:38:42 UTC (1325489922) 2012-01-02 16:38:42 CJT (1325489922)",
|
|
"2012-01-02 21:51:08 UTC (1325541068) 2012-01-03 06:51:08 CJT (1325541068)",
|
|
"2012-01-03 00:00:00 UTC (1325548800) 2012-01-03 09:00:00 CJT (1325548800)",
|
|
),
|
|
("lat: -33.93 lon: 18.42 cape town first: night",
|
|
"2012-01-02 00:00:00 UTC (1325462400) 2012-01-02 02:00:00 SAST (1325462400)",
|
|
"2012-01-02 03:39:13 UTC (1325475553) 2012-01-02 05:39:13 SAST (1325475553)",
|
|
"2012-01-02 18:00:57 UTC (1325527257) 2012-01-02 20:00:57 SAST (1325527257)",
|
|
"2012-01-03 00:00:00 UTC (1325548800) 2012-01-03 02:00:00 SAST (1325548800)",
|
|
),
|
|
("lat: 51.4791 lon: 0 greenwich first: night",
|
|
"2012-01-02 00:00:00 UTC (1325462400) 2012-01-02 00:00:00 GMT (1325462400)",
|
|
"2012-01-02 08:05:18 UTC (1325491518) 2012-01-02 08:05:18 GMT (1325491518)",
|
|
"2012-01-02 16:02:17 UTC (1325520137) 2012-01-02 16:02:17 GMT (1325520137)",
|
|
"2012-01-03 00:00:00 UTC (1325548800) 2012-01-03 00:00:00 GMT (1325548800)",
|
|
),
|
|
("lat: 42.358 lon: -71.06 boston first: night",
|
|
"2012-01-02 00:00:00 UTC (1325462400) 2012-01-01 19:00:00 EST (1325462400)",
|
|
"2012-01-02 12:13:26 UTC (1325506406) 2012-01-02 07:13:26 EST (1325506406)",
|
|
"2012-01-02 21:22:49 UTC (1325539369) 2012-01-02 16:22:49 EST (1325539369)",
|
|
"2012-01-03 00:00:00 UTC (1325548800) 2012-01-02 19:00:00 EST (1325548800)",
|
|
),
|
|
("lat: 21.3 lon: -157.8167 honolulu first: day",
|
|
"2012-01-02 00:00:00 UTC (1325462400) 2012-01-01 14:00:00 HST (1325462400)",
|
|
"2012-01-02 04:00:45 UTC (1325476845) 2012-01-01 18:00:45 HST (1325476845)",
|
|
"2012-01-02 17:09:09 UTC (1325524149) 2012-01-02 07:09:09 HST (1325524149)",
|
|
"2012-01-03 00:00:00 UTC (1325548800) 2012-01-02 14:00:00 HST (1325548800)",
|
|
)),
|
|
(
|
|
("lat: -33.86 lon: 151.21 sydney first: day",
|
|
"2012-01-02 22:00:00 UTC (1325541600) 2012-01-03 09:00:00 EST (1325541600)",
|
|
"2012-01-03 09:09:43 UTC (1325581783) 2012-01-03 20:09:43 EST (1325581783)",
|
|
"2012-01-03 18:49:31 UTC (1325616571) 2012-01-04 05:49:31 EST (1325616571)",
|
|
"2012-01-03 22:00:00 UTC (1325628000) 2012-01-04 09:00:00 EST (1325628000)",
|
|
),
|
|
("lat: 35.6895 lon: 139.6917 tokyo first: day",
|
|
"2012-01-02 22:00:00 UTC (1325541600) 2012-01-03 07:00:00 CJT (1325541600)",
|
|
"2012-01-03 07:39:29 UTC (1325576369) 2012-01-03 16:39:29 CJT (1325576369)",
|
|
"2012-01-03 21:51:16 UTC (1325627476) 2012-01-04 06:51:16 CJT (1325627476)",
|
|
"2012-01-03 22:00:00 UTC (1325628000) 2012-01-04 07:00:00 CJT (1325628000)",
|
|
),
|
|
("lat: -33.93 lon: 18.42 cape town first: night",
|
|
"2012-01-02 22:00:00 UTC (1325541600) 2012-01-03 00:00:00 SAST (1325541600)",
|
|
"2012-01-03 03:39:59 UTC (1325561999) 2012-01-03 05:39:59 SAST (1325561999)",
|
|
"2012-01-03 18:01:07 UTC (1325613667) 2012-01-03 20:01:07 SAST (1325613667)",
|
|
"2012-01-03 22:00:00 UTC (1325628000) 2012-01-04 00:00:00 SAST (1325628000)",
|
|
),
|
|
("lat: 51.4791 lon: 0 greenwich first: night",
|
|
"2012-01-02 22:00:00 UTC (1325541600) 2012-01-02 22:00:00 GMT (1325541600)",
|
|
"2012-01-03 08:05:09 UTC (1325577909) 2012-01-03 08:05:09 GMT (1325577909)",
|
|
"2012-01-03 16:03:22 UTC (1325606602) 2012-01-03 16:03:22 GMT (1325606602)",
|
|
"2012-01-03 22:00:00 UTC (1325628000) 2012-01-03 22:00:00 GMT (1325628000)",
|
|
),
|
|
("lat: 42.358 lon: -71.06 boston first: night",
|
|
"2012-01-02 22:00:00 UTC (1325541600) 2012-01-02 17:00:00 EST (1325541600)",
|
|
"2012-01-03 12:13:28 UTC (1325592808) 2012-01-03 07:13:28 EST (1325592808)",
|
|
"2012-01-03 21:23:42 UTC (1325625822) 2012-01-03 16:23:42 EST (1325625822)",
|
|
"2012-01-03 22:00:00 UTC (1325628000) 2012-01-03 17:00:00 EST (1325628000)",
|
|
),
|
|
("lat: 21.3 lon: -157.8167 honolulu first: day",
|
|
"2012-01-02 22:00:00 UTC (1325541600) 2012-01-02 12:00:00 HST (1325541600)",
|
|
"2012-01-03 04:01:22 UTC (1325563282) 2012-01-02 18:01:22 HST (1325563282)",
|
|
"2012-01-03 17:09:26 UTC (1325610566) 2012-01-03 07:09:26 HST (1325610566)",
|
|
"2012-01-03 22:00:00 UTC (1325628000) 2012-01-03 12:00:00 HST (1325628000)"
|
|
))
|
|
]
|
|
for i,t in enumerate(times):
|
|
for j,l in enumerate(locs):
|
|
os.environ['TZ'] = l[3]
|
|
first, values = getDayNightTransitions(t[0], t[1], l[0], l[1])
|
|
self.assertEqual("lat: %s lon: %s %s first: %s" %
|
|
(l[0], l[1], l[2], first),
|
|
expected[i][j][0])
|
|
self.assertEqual("%s %s" % (timestamp_to_gmtime(t[0]),
|
|
timestamp_to_string(t[0])),
|
|
expected[i][j][1])
|
|
self.assertEqual("%s %s" % (timestamp_to_gmtime(values[0]),
|
|
timestamp_to_string(values[0])),
|
|
expected[i][j][2])
|
|
self.assertEqual("%s %s" % (timestamp_to_gmtime(values[1]),
|
|
timestamp_to_string(values[1])),
|
|
expected[i][j][3])
|
|
self.assertEqual("%s %s" % (timestamp_to_gmtime(t[1]),
|
|
timestamp_to_string(t[1])),
|
|
expected[i][j][4])
|
|
os.environ['TZ'] = tz
|
|
|
|
if __name__ == '__main__':
|
|
os.environ['TZ'] = 'America/Los_Angeles'
|
|
unittest.main()
|