Current File : //usr/share/doc/python3-hamcrest/examples/CustomDateMatcher.py
import sys
sys.path.append("..")
from hamcrest.core.base_matcher import BaseMatcher
from hamcrest.core.helpers.hasmethod import hasmethod
from hamcrest import *
import unittest
import datetime
class IsGivenDayOfWeek(BaseMatcher):
"""Matches dates that fall on a given day of the week."""
def __init__(self, day):
self.day = day # Monday is 0, Sunday is 6
def _matches(self, item):
"""Test whether item matches."""
if not hasmethod(item, "weekday"):
return False
return item.weekday() == self.day
def describe_to(self, description):
"""Describe the matcher."""
day_as_string = [
"Monday",
"Tuesday",
"Wednesday",
"Thursday",
"Friday",
"Saturday",
"Sunday",
]
description.append_text("calendar date falling on ").append_text(day_as_string[self.day])
def on_a_saturday():
"""Factory function to generate Saturday matcher."""
return IsGivenDayOfWeek(5)
class SampleTest(unittest.TestCase):
def testDateIsOnASaturday(self):
"""Example of successful match."""
d = datetime.date(2008, 4, 26)
assert_that(d, is_(on_a_saturday()))
def testFailsWithMismatchedDate(self):
"""Example of what happens with date that doesn't match."""
d = datetime.date(2008, 4, 6)
assert_that(d, is_(on_a_saturday()))
def testFailsWithNonDate(self):
"""Example of what happens with object that isn't a date."""
d = "oops"
assert_that(d, is_(on_a_saturday()))
if __name__ == "__main__":
unittest.main()
Mini Shell