test/test_dates.py

   1 #
   2 # Copyright (c) 2001 Bizar Software Pty Ltd (http://www.bizarsoftware.com.au/)
   3 # This module is free software, and you may redistribute it and/or modify
   4 # under the same terms as Python, so long as this copyright message and
   5 # disclaimer are retained in their original form.
   6 #
   7 # IN NO EVENT SHALL BIZAR SOFTWARE PTY LTD BE LIABLE TO ANY PARTY FOR
   8 # DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING
   9 # OUT OF THE USE OF THIS CODE, EVEN IF THE AUTHOR HAS BEEN ADVISED OF THE
  10 # POSSIBILITY OF SUCH DAMAGE.
  11 #
  12 # BIZAR SOFTWARE PTY LTD SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING,
  13 # BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  14 # FOR A PARTICULAR PURPOSE.  THE CODE PROVIDED HEREUNDER IS ON AN "AS IS"
  15 # BASIS, AND THERE IS NO OBLIGATION WHATSOEVER TO PROVIDE MAINTENANCE,
  16 # SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
  17 #
  18 # $Id: test_dates.py,v 1.31 2003-12-04 23:06:53 richard Exp $
  19 from __future__ import nested_scopes
  20
  21 import unittest, time
  22
  23 from roundup.date import Date, Interval, Range, fixTimeOverflow
  24
  25 class DateTestCase(unittest.TestCase):
  26     def testDateInterval(self):
  27         ae = self.assertEqual
  28         date = Date("2000-06-26.00:34:02 + 2d")
  29         ae(str(date), '2000-06-28.00:34:02')
  30         date = Date("2000-02-27 + 2d")
  31         ae(str(date), '2000-02-29.00:00:00')
  32         date = Date("2001-02-27 + 2d")
  33         ae(str(date), '2001-03-01.00:00:00')
  34
  35     def testDate(self):
  36         ae = self.assertEqual
  37         date = Date("2000-04-17")
  38         ae(str(date), '2000-04-17.00:00:00')
  39         date = Date("2000/04/17")
  40         ae(str(date), '2000-04-17.00:00:00')
  41         date = Date("2000-4-7")
  42         ae(str(date), '2000-04-07.00:00:00')
  43         date = Date("2000-4-17")
  44         ae(str(date), '2000-04-17.00:00:00')
  45         date = Date("01-25")
  46         y, m, d, x, x, x, x, x, x = time.gmtime(time.time())
  47         ae(str(date), '%s-01-25.00:00:00'%y)
  48         date = Date("2000-04-17.03:45")
  49         ae(str(date), '2000-04-17.03:45:00')
  50         date = Date("2000/04/17.03:45")
  51         ae(str(date), '2000-04-17.03:45:00')
  52         date = Date("08-13.22:13")
  53         ae(str(date), '%s-08-13.22:13:00'%y)
  54         date = Date("11-07.09:32:43")
  55         ae(str(date), '%s-11-07.09:32:43'%y)
  56         date = Date("14:25")
  57         ae(str(date), '%s-%02d-%02d.14:25:00'%(y, m, d))
  58         date = Date("8:47:11")
  59         ae(str(date), '%s-%02d-%02d.08:47:11'%(y, m, d))
  60         ae(str(Date('2003')), '2003-01-01.00:00:00')
  61         ae(str(Date('2004-06')), '2004-06-01.00:00:00')
  62
  63     def testDateError(self):
  64         self.assertRaises(ValueError, Date, "12")
  65
  66     def testOffset(self):
  67         ae = self.assertEqual
  68         date = Date("2000-04-17", -5)
  69         ae(str(date), '2000-04-17.05:00:00')
  70         date = Date("01-25", -5)
  71         y, m, d, x, x, x, x, x, x = time.gmtime(time.time())
  72         ae(str(date), '%s-01-25.05:00:00'%y)
  73         date = Date("2000-04-17.03:45", -5)
  74         ae(str(date), '2000-04-17.08:45:00')
  75         date = Date("08-13.22:13", -5)
  76         ae(str(date), '%s-08-14.03:13:00'%y)
  77         date = Date("11-07.09:32:43", -5)
  78         ae(str(date), '%s-11-07.14:32:43'%y)
  79         date = Date("14:25", -5)
  80         ae(str(date), '%s-%02d-%02d.19:25:00'%(y, m, d))
  81         date = Date("8:47:11", -5)
  82         ae(str(date), '%s-%02d-%02d.13:47:11'%(y, m, d))
  83
  84     def testOffsetRandom(self):
  85         ae = self.assertEqual
  86         # XXX unsure of the usefulness of these, they're pretty random
  87         date = Date('2000-01-01') + Interval('- 2y 2m')
  88         ae(str(date), '1997-11-01.00:00:00')
  89         date = Date('2000-01-01 - 2y 2m')
  90         ae(str(date), '1997-11-01.00:00:00')
  91         date = Date('2000-01-01') + Interval('2m')
  92         ae(str(date), '2000-03-01.00:00:00')
  93         date = Date('2000-01-01 + 2m')
  94         ae(str(date), '2000-03-01.00:00:00')
  95
  96         date = Date('2000-01-01') + Interval('60d')
  97         ae(str(date), '2000-03-01.00:00:00')
  98         date = Date('2001-01-01') + Interval('60d')
  99         ae(str(date), '2001-03-02.00:00:00')
 100
 101     def testOffsetAdd(self):
 102         ae = self.assertEqual
 103         date = Date('2000-02-28.23:59:59') + Interval('00:00:01')
 104         ae(str(date), '2000-02-29.00:00:00')
 105         date = Date('2001-02-28.23:59:59') + Interval('00:00:01')
 106         ae(str(date), '2001-03-01.00:00:00')
 107
 108         date = Date('2000-02-28.23:58:59') + Interval('00:01:01')
 109         ae(str(date), '2000-02-29.00:00:00')
 110         date = Date('2001-02-28.23:58:59') + Interval('00:01:01')
 111         ae(str(date), '2001-03-01.00:00:00')
 112
 113         date = Date('2000-02-28.22:58:59') + Interval('01:01:01')
 114         ae(str(date), '2000-02-29.00:00:00')
 115         date = Date('2001-02-28.22:58:59') + Interval('01:01:01')
 116         ae(str(date), '2001-03-01.00:00:00')
 117
 118         date = Date('2000-02-28.22:58:59') + Interval('00:00:3661')
 119         ae(str(date), '2000-02-29.00:00:00')
 120         date = Date('2001-02-28.22:58:59') + Interval('00:00:3661')
 121         ae(str(date), '2001-03-01.00:00:00')
 122
 123     def testOffsetSub(self):
 124         ae = self.assertEqual
 125         date = Date('2000-12-01') - Interval('- 1d')
 126
 127         date = Date('2000-01-01') - Interval('- 2y 2m')
 128         ae(str(date), '2002-03-01.00:00:00')
 129         date = Date('2000-01-01') - Interval('2m')
 130         ae(str(date), '1999-11-01.00:00:00')
 131
 132         date = Date('2000-03-01') - Interval('60d')
 133         ae(str(date), '2000-01-01.00:00:00')
 134         date = Date('2001-03-02') - Interval('60d')
 135         ae(str(date), '2001-01-01.00:00:00')
 136
 137         date = Date('2000-02-29.00:00:00') - Interval('00:00:01')
 138         ae(str(date), '2000-02-28.23:59:59')
 139         date = Date('2001-03-01.00:00:00') - Interval('00:00:01')
 140         ae(str(date), '2001-02-28.23:59:59')
 141
 142         date = Date('2000-02-29.00:00:00') - Interval('00:01:01')
 143         ae(str(date), '2000-02-28.23:58:59')
 144         date = Date('2001-03-01.00:00:00') - Interval('00:01:01')
 145         ae(str(date), '2001-02-28.23:58:59')
 146
 147         date = Date('2000-02-29.00:00:00') - Interval('01:01:01')
 148         ae(str(date), '2000-02-28.22:58:59')
 149         date = Date('2001-03-01.00:00:00') - Interval('01:01:01')
 150         ae(str(date), '2001-02-28.22:58:59')
 151
 152         date = Date('2000-02-29.00:00:00') - Interval('00:00:3661')
 153         ae(str(date), '2000-02-28.22:58:59')
 154         date = Date('2001-03-01.00:00:00') - Interval('00:00:3661')
 155         ae(str(date), '2001-02-28.22:58:59')
 156
 157     def testDateLocal(self):
 158         ae = self.assertEqual
 159         date = Date("02:42:20")
 160         date = date.local(10)
 161         y, m, d, x, x, x, x, x, x = time.gmtime(time.time())
 162         ae(str(date), '%s-%02d-%02d.12:42:20'%(y, m, d))
 163
 164     def testIntervalInit(self):
 165         ae = self.assertEqual
 166         ae(str(Interval('3y')), '+ 3y')
 167         ae(str(Interval('2 y 1 m')), '+ 2y 1m')
 168         ae(str(Interval('1m 25d')), '+ 1m 25d')
 169         ae(str(Interval('-2w 3 d ')), '- 17d')
 170         ae(str(Interval(' - 1 d 2:50 ')), '- 1d 2:50')
 171         ae(str(Interval(' 14:00 ')), '+ 14:00')
 172         ae(str(Interval(' 0:04:33 ')), '+ 0:04:33')
 173
 174     def testIntervalInitDate(self):
 175         ae = self.assertEqual
 176         now = Date('.')
 177         now.hour = now.minute = now.second = 0
 178         then = now + Interval('2d')
 179         ae((Interval(str(then))), Interval('- 2d'))
 180         then = now - Interval('2d')
 181         ae(Interval(str(then)), Interval('+ 2d'))
 182
 183     def testIntervalAddMonthBoundary(self):
 184         # force the transition over a month boundary
 185         now = Date('2003-10-30.00:00:00')
 186         then = now + Interval('2d')
 187         self.assertEqual(str(then), '2003-11-01.00:00:00')
 188         now = Date('2004-02-28.00:00:00')
 189         then = now + Interval('1d')
 190         self.assertEqual(str(then), '2004-02-29.00:00:00')
 191         now = Date('2003-02-28.00:00:00')
 192         then = now + Interval('1d')
 193         self.assertEqual(str(then), '2003-03-01.00:00:00')
 194         now = Date('2003-01-01.00:00:00')
 195         then = now + Interval('59d')
 196         self.assertEqual(str(then), '2003-03-01.00:00:00')
 197         now = Date('2004-01-01.00:00:00')
 198         then = now + Interval('59d')
 199         self.assertEqual(str(then), '2004-02-29.00:00:00')
 200
 201     def testIntervalSubtractMonthBoundary(self):
 202         # force the transition over a month boundary
 203         now = Date('2003-11-01.00:00:00')
 204         then = now - Interval('2d')
 205         self.assertEqual(str(then), '2003-10-30.00:00:00')
 206         now = Date('2004-02-29.00:00:00')
 207         then = now - Interval('1d')
 208         self.assertEqual(str(then), '2004-02-28.00:00:00')
 209         now = Date('2003-03-01.00:00:00')
 210         then = now - Interval('1d')
 211         self.assertEqual(str(then), '2003-02-28.00:00:00')
 212         now = Date('2003-03-01.00:00:00')
 213         then = now - Interval('59d')
 214         self.assertEqual(str(then), '2003-01-01.00:00:00')
 215         now = Date('2004-02-29.00:00:00')
 216         then = now - Interval('59d')
 217         self.assertEqual(str(then), '2004-01-01.00:00:00')
 218
 219     def testIntervalAddYearBoundary(self):
 220         # force the transition over a year boundary
 221         now = Date('2003-12-30.00:00:00')
 222         then = now + Interval('2d')
 223         self.assertEqual(str(then), '2004-01-01.00:00:00')
 224         now = Date('2003-01-01.00:00:00')
 225         then = now + Interval('365d')
 226         self.assertEqual(str(then), '2004-01-01.00:00:00')
 227         now = Date('2004-01-01.00:00:00')
 228         then = now + Interval('366d')
 229         self.assertEqual(str(then), '2005-01-01.00:00:00')
 230
 231     def testIntervalSubtractYearBoundary(self):
 232         # force the transition over a year boundary
 233         now = Date('2003-01-01.00:00:00')
 234         then = now - Interval('2d')
 235         self.assertEqual(str(then), '2002-12-30.00:00:00')
 236         now = Date('2004-02-01.00:00:00')
 237         then = now - Interval('365d')
 238         self.assertEqual(str(then), '2003-02-01.00:00:00')
 239         now = Date('2005-02-01.00:00:00')
 240         then = now - Interval('365d')
 241         self.assertEqual(str(then), '2004-02-02.00:00:00')
 242
 243     def testDateSubtract(self):
 244         # These are thoroughly broken right now.
 245         i = Date('2003-03-15.00:00:00') - Date('2003-03-10.00:00:00')
 246         self.assertEqual(i, Interval('5d'))
 247         i = Date('2003-02-01.00:00:00') - Date('2003-03-01.00:00:00')
 248         self.assertEqual(i, Interval('-28d'))
 249         i = Date('2003-03-01.00:00:00') - Date('2003-02-01.00:00:00')
 250         self.assertEqual(i, Interval('28d'))
 251         i = Date('2003-03-03.00:00:00') - Date('2003-02-01.00:00:00')
 252         self.assertEqual(i, Interval('30d'))
 253         i = Date('2003-03-03.00:00:00') - Date('2002-02-01.00:00:00')
 254         self.assertEqual(i, Interval('395d'))
 255         i = Date('2003-03-03.00:00:00') - Date('2003-04-01.00:00:00')
 256         self.assertEqual(i, Interval('-29d'))
 257         i = Date('2003-03-01.00:00:00') - Date('2003-02-01.00:00:00')
 258         self.assertEqual(i, Interval('28d'))
 259         # force the transition over a year boundary
 260         i = Date('2003-01-01.00:00:00') - Date('2002-01-01.00:00:00')
 261         self.assertEqual(i, Interval('365d'))
 262
 263     def testIntervalAdd(self):
 264         ae = self.assertEqual
 265         ae(str(Interval('1y') + Interval('1y')), '+ 2y')
 266         ae(str(Interval('1y') + Interval('1m')), '+ 1y 1m')
 267         ae(str(Interval('1y') + Interval('2:40')), '+ 1y 2:40')
 268         ae(str(Interval('1y') + Interval('- 1y')), '')
 269         ae(str(Interval('- 1y') + Interval('1y')), '')
 270         ae(str(Interval('- 1y') + Interval('- 1y')), '- 2y')
 271         ae(str(Interval('1y') + Interval('- 1m')), '+ 11m')
 272         ae(str(Interval('1:00') + Interval('1:00')), '+ 2:00')
 273         ae(str(Interval('0:50') + Interval('0:50')), '+ 1:40')
 274         ae(str(Interval('1:50') + Interval('- 1:50')), '')
 275         ae(str(Interval('- 1:50') + Interval('1:50')), '')
 276         ae(str(Interval('- 1:50') + Interval('- 1:50')), '- 3:40')
 277         ae(str(Interval('1:59:59') + Interval('00:00:01')), '+ 2:00')
 278         ae(str(Interval('2:00') + Interval('- 00:00:01')), '+ 1:59:59')
 279
 280     def testIntervalSub(self):
 281         ae = self.assertEqual
 282         ae(str(Interval('1y') - Interval('- 1y')), '+ 2y')
 283         ae(str(Interval('1y') - Interval('- 1m')), '+ 1y 1m')
 284         ae(str(Interval('1y') - Interval('- 2:40')), '+ 1y 2:40')
 285         ae(str(Interval('1y') - Interval('1y')), '')
 286         ae(str(Interval('1y') - Interval('1m')), '+ 11m')
 287         ae(str(Interval('1:00') - Interval('- 1:00')), '+ 2:00')
 288         ae(str(Interval('0:50') - Interval('- 0:50')), '+ 1:40')
 289         ae(str(Interval('1:50') - Interval('1:50')), '')
 290         ae(str(Interval('1:59:59') - Interval('- 00:00:01')), '+ 2:00')
 291         ae(str(Interval('2:00') - Interval('00:00:01')), '+ 1:59:59')
 292
 293     def testOverflow(self):
 294         ae = self.assertEqual
 295         ae(fixTimeOverflow((1,0,0,0, 0, 0, 60)), (1,0,0,0, 0, 1, 0))
 296         ae(fixTimeOverflow((1,0,0,0, 0, 0, 100)), (1,0,0,0, 0, 1, 40))
 297         ae(fixTimeOverflow((1,0,0,0, 0, 0, 60*60)), (1,0,0,0, 1, 0, 0))
 298         ae(fixTimeOverflow((1,0,0,0, 0, 0, 24*60*60)), (1,0,0,1, 0, 0, 0))
 299         ae(fixTimeOverflow((1,0,0,0, 0, 0, -1)), (-1,0,0,0, 0, 0, 1))
 300         ae(fixTimeOverflow((1,0,0,0, 0, 0, -100)), (-1,0,0,0, 0, 1, 40))
 301         ae(fixTimeOverflow((1,0,0,0, 0, 0, -60*60)), (-1,0,0,0, 1, 0, 0))
 302         ae(fixTimeOverflow((1,0,0,0, 0, 0, -24*60*60)), (-1,0,0,1, 0, 0, 0))
 303         ae(fixTimeOverflow((-1,0,0,0, 0, 0, 1)), (-1,0,0,0, 0, 0, 1))
 304         ae(fixTimeOverflow((-1,0,0,0, 0, 0, 100)), (-1,0,0,0, 0, 1, 40))
 305         ae(fixTimeOverflow((-1,0,0,0, 0, 0, 60*60)), (-1,0,0,0, 1, 0, 0))
 306         ae(fixTimeOverflow((-1,0,0,0, 0, 0, 24*60*60)), (-1,0,0,1, 0, 0, 0))
 307
 308     def testDivision(self):
 309         ae = self.assertEqual
 310         ae(str(Interval('1y')/2), '+ 6m')
 311         ae(str(Interval('1:00')/2), '+ 0:30')
 312         ae(str(Interval('00:01')/2), '+ 0:00:30')
 313
 314     def testSorting(self):
 315         ae = self.assertEqual
 316         i1 = Interval('1y')
 317         i2 = Interval('1d')
 318         l = [i1, i2]; l.sort()
 319         ae(l, [i2, i1])
 320         l = [i2, i1]; l.sort()
 321         ae(l, [i2, i1])
 322         i1 = Interval('- 2d')
 323         i2 = Interval('1d')
 324         l = [i1, i2]; l.sort()
 325         ae(l, [i1, i2])
 326
 327         i1 = Interval("1:20")
 328         i2 = Interval("2d")
 329         i3 = Interval("3:30")
 330         l = [i1, i2, i3]; l.sort()
 331         ae(l, [i1, i3, i2])
 332
 333     def testGranularity(self):
 334         ae = self.assertEqual
 335         ae(str(Date('2003-2-12', add_granularity=1)), '2003-02-12.23:59:59')
 336         ae(str(Date('2003-1-1.23:00', add_granularity=1)), '2003-01-01.23:00:59')
 337         ae(str(Date('2003', add_granularity=1)), '2003-12-31.23:59:59')
 338         ae(str(Date('2003-5', add_granularity=1)), '2003-05-31.23:59:59')
 339         ae(str(Interval('+1w', add_granularity=1)), '+ 14d')
 340         ae(str(Interval('-2m 3w', add_granularity=1)), '- 2m 14d')
 341
 342     def testIntervalPretty(self):
 343         def ae(spec, pretty):
 344             self.assertEqual(Interval(spec).pretty(), pretty)
 345         ae('2y', 'in 2 years')
 346         ae('1y', 'in 1 year')
 347         ae('2m', 'in 2 months')
 348         ae('1m 30d', 'in 2 months')
 349         ae('60d', 'in 2 months')
 350         ae('59d', 'in 1 month')
 351         ae('1m', 'in 1 month')
 352         ae('29d', 'in 1 month')
 353         ae('28d', 'in 4 weeks')
 354         ae('8d', 'in 1 week')
 355         ae('7d', 'in 7 days')
 356         ae('1w', 'in 7 days')
 357         ae('2d', 'in 2 days')
 358         ae('1d', 'tomorrow')
 359         ae('02:00:00', 'in 2 hours')
 360         ae('01:59:00', 'in 1 3/4 hours')
 361         ae('01:45:00', 'in 1 3/4 hours')
 362         ae('01:30:00', 'in 1 1/2 hours')
 363         ae('01:29:00', 'in 1 1/4 hours')
 364         ae('01:00:00', 'in an hour')
 365         ae('00:30:00', 'in 1/2 an hour')
 366         ae('00:15:00', 'in 1/4 hour')
 367         ae('00:02:00', 'in 2 minutes')
 368         ae('00:01:00', 'in 1 minute')
 369         ae('00:00:30', 'in a moment')
 370         ae('-00:00:30', 'just now')
 371         ae('-1d', 'yesterday')
 372         ae('-1y', '1 year ago')
 373         ae('-2y', '2 years ago')
 374
 375 def test_suite():
 376     suite = unittest.TestSuite()
 377     suite.addTest(unittest.makeSuite(DateTestCase))
 378     return suite
 379
 380 if __name__ == '__main__':
 381     runner = unittest.TextTestRunner()
 382     unittest.main(testRunner=runner)
 383
 384 # vim: set filetype=python ts=4 sw=4 et si