Source code for uw.local.termtools_test

import unittest

from uw.local.termtools import *

class TermtoolsTests (unittest.TestCase):

    def setUp (self):
        """ Setup method which runs before each test case.
        """
        self.winter_2019 = make (2019, 1)
        self.spring_2019 = make (2019, 5)
        self.fall_2019 = make (2019, 9)
        self.winter_2020 = make (2020, 1)
        self.spring_2020 = make (2020, 5)
        self.fall_2020 = make (2020, 9)

    def test_year (self):
        """ Tests if the year () method works as expected.
        """
        self.assertEqual (2019, self.winter_2019.year ())
        self.assertEqual (2019, self.spring_2019.year ())
        self.assertEqual (2019, self.fall_2019.year ())

    def test_month (self):
        """ Tests if the month () method works as expected.
        """
        self.assertEqual (1, self.winter_2019.month ())
        self.assertEqual (5, self.spring_2019.month ())
        self.assertEqual (9, self.fall_2019.month ())

    def test_season_name (self):
        """ Tests if the seasonName () method works as expected.
        """
        self.assertEqual ('Winter', self.winter_2019.seasonName ())
        self.assertEqual ('Spring', self.spring_2019.seasonName ())
        self.assertEqual ('Fall', self.fall_2019.seasonName ())

    def test_season_initial (self):
        """ Tests if the seasonInitial () method works as expected.
        """
        self.assertEqual ('W', self.winter_2019.seasonInitial ())
        self.assertEqual ('S', self.spring_2019.seasonInitial ())
        self.assertEqual ('F', self.fall_2019.seasonInitial ())

    def test_description (self):
        """ Tests if the description () method works as expected.
        """
        self.assertEqual ('Winter 2019', self.winter_2019.description ())
        self.assertEqual ('Spring 2019', self.spring_2019.description ())
        self.assertEqual ('Fall 2019', self.fall_2019.description ())

    def test_abbreviation (self):
        """ Tests if the abbreviation () method works as expected.
        """
        self.assertEqual ('W19', self.winter_2019.abbreviation ())
        self.assertEqual ('S19', self.spring_2019.abbreviation ())
        self.assertEqual ('F19', self.fall_2019.abbreviation ())

    def test_code (self):
        """ Tests if the code () method works as expected.
        """
        self.assertEqual ('1191', self.winter_2019.code ())
        self.assertEqual ('1195', self.spring_2019.code ())
        self.assertEqual ('1199', self.fall_2019.code ())

    def test_numeric_code (self):
        """ Tests if the numeric () method works as expected.
        """
        self.assertEqual (1191, self.winter_2019.numericCode ())
        self.assertEqual (1195, self.spring_2019.numericCode ())
        self.assertEqual (1199, self.fall_2019.numericCode ())

    def test_next_term (self):
        """ Tests if the next_term () method works as expected.
        """
        self.assertEqual (self.spring_2019, self.winter_2019.nextTerm ())
        self.assertEqual (self.fall_2019, self.spring_2019.nextTerm ())
        self.assertEqual (self.winter_2020, self.fall_2019.nextTerm ())

    def test_prev_term (self):
        """ Tests if the prev_term () method works as expected.
        """
        self.assertEqual (self.spring_2019.prevTerm (), self.winter_2019)
        self.assertEqual (self.fall_2019.prevTerm (), self.spring_2019)
        self.assertEqual (self.winter_2020.prevTerm (), self.fall_2019)

    def test_next_year (self):
        """ Tests if the next_year () method works as expected.
        """
        self.assertEqual (self.winter_2020, self.winter_2019.nextYear ())
        self.assertEqual (self.spring_2020, self.spring_2019.nextYear ())
        self.assertEqual (self.fall_2020, self.fall_2019.nextYear ())

    def test_start_date (self):
        """ Tests if the start_date () method works as expected.
        """
        self.assertEqual (datetime.date (2019, 1, 1), self.winter_2019.startDate ())
        self.assertEqual (datetime.date (2019, 5, 1), self.spring_2019.startDate ())
        self.assertEqual (datetime.date (2019, 9, 1), self.fall_2019.startDate ())

    def test_make_returns_correct_term (self):
        """ Tests if the make () function works when given a valid input.
        """
        self.assertEqual (self.winter_2019, make (2019, 1))
        self.assertEqual (self.spring_2019, make (2019, 5))
        self.assertEqual (self.fall_2019, make (2019, 9))

    def test_make_throws_value_error (self):
        """ Tests if the make () function throws an error when given an
            invalid input.
        """
        self.assertRaises (ValueError, make, 2019, 0)
        self.assertRaises (ValueError, make, 2019, 2)
        self.assertRaises (ValueError, make, 2019, 3)
        self.assertRaises (ValueError, make, 2019, 4)
        self.assertRaises (ValueError, make, 2019, 6)
        self.assertRaises (ValueError, make, 2019, 7)
        self.assertRaises (ValueError, make, 2019, 8)
        self.assertRaises (ValueError, make, 2019, 10)
        self.assertRaises (ValueError, make, 2019, 11)
        self.assertRaises (ValueError, make, 2019, 12)
        self.assertRaises (ValueError, make, 2019, 13)

    def test_from_date (self):
        """ Tests if the from_date () function works as expected.
        """
        self.assertEqual (self.winter_2019, fromDate (datetime.date (2019, 1, 1)))
        self.assertEqual (self.winter_2019, fromDate (datetime.date (2019, 2, 1)))
        self.assertEqual (self.winter_2019, fromDate (datetime.date (2019, 3, 1)))
        self.assertEqual (self.winter_2019, fromDate (datetime.date (2019, 4, 1)))

    def test_from_code (self):
        """ Tests if the from_code () function works as expected.
        """
        self.assertEqual (self.winter_2019, fromCode (1191))
        self.assertEqual (self.spring_2019, fromCode (1195))
        self.assertEqual (self.fall_2019, fromCode (1199))