Skip to content
This repository has been archived by the owner on Nov 5, 2021. It is now read-only.

Commit

Permalink
Fix tests after date parsing #18
Browse files Browse the repository at this point in the history
  • Loading branch information
tomislacker committed Jun 7, 2018
1 parent 36a6231 commit 777e180
Showing 1 changed file with 42 additions and 8 deletions.
50 changes: 42 additions & 8 deletions tests/test_parsing.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,9 @@
from datetime import timedelta
from dateutil import tz
from nose2.tools import such

import dateutil.parser as dateutil_parser

from mkerefuse.refuse import RefusePickup


Expand Down Expand Up @@ -38,16 +42,31 @@ def test(case):
def test(case):
case.assertEqual(
it.parser.next_pickup_garbage.isoformat(),
'2016-12-27T00:00:00')
(
dateutil_parser.parse('TUESDAY DECEMBER 27, 2016')
.replace(tzinfo=tz.gettz(RefusePickup.pickup_tz))
+ timedelta(**RefusePickup.pickup_offset)
).isoformat(),
)

@it.should('have the correct next recycle pickup range')
def test(case):
case.assertEqual(
it.parser.next_pickup_recycle_after.isoformat(),
'2017-01-03T00:00:00')
case.assertEqual(
it.parser.next_pickup_recycle_before.isoformat(),
'2017-01-04T00:00:00')
(
dateutil_parser.parse('WEDNESDAY JANUARY 4 2017')
.replace(tzinfo=tz.gettz(RefusePickup.pickup_tz))
+ timedelta(**RefusePickup.pickup_offset)
).isoformat(),
)
case.assertEqual(
it.parser.next_pickup_recycle_after.isoformat(),
(
dateutil_parser.parse('TUESDAY JANUARY 3 2017')
.replace(tzinfo=tz.gettz(RefusePickup.pickup_tz))
+ timedelta(**RefusePickup.pickup_offset)
).isoformat(),
)

with it.having('non-garbage day, unknown recycling, in Winter'):
@it.has_setup
Expand All @@ -71,7 +90,12 @@ def test(case):
def test(case):
case.assertEqual(
it.parser.next_pickup_garbage.isoformat(),
'2016-12-29T00:00:00')
(
dateutil_parser.parse('THURSDAY DECEMBER 29, 2016')
.replace(tzinfo=tz.gettz(RefusePickup.pickup_tz))
+ timedelta(**RefusePickup.pickup_offset)
).isoformat(),
)

@it.should('have the correct next recycle pickup range')
def test(case):
Expand Down Expand Up @@ -108,13 +132,23 @@ def test(case):
def test(case):
case.assertEqual(
it.parser.next_pickup_garbage.isoformat(),
'2017-04-14T00:00:00')
(
dateutil_parser.parse('FRIDAY APRIL 14, 2017')
.replace(tzinfo=tz.gettz(RefusePickup.pickup_tz))
+ timedelta(**RefusePickup.pickup_offset)
).isoformat(),
)

@it.should('have the correct next recycle pickup')
def test(case):
case.assertEqual(
it.parser.next_pickup_recycle.isoformat(),
'2017-04-12T00:00:00')
(
dateutil_parser.parse('WEDNESDAY APRIL 12, 2017')
.replace(tzinfo=tz.gettz(RefusePickup.pickup_tz))
+ timedelta(**RefusePickup.pickup_offset)
).isoformat(),
)

# Recycle ranges (for Winter) should be empty
case.assertEqual(
Expand Down

0 comments on commit 777e180

Please sign in to comment.