Skip to content

Commit

Permalink
add unit tests for util module
Browse files Browse the repository at this point in the history
  • Loading branch information
@ebridges
ebridges committed Feb 10, 2018
1 parent 3e1f275 commit c0d0eab
Show file tree
Hide file tree
Showing 2 changed files with 180 additions and 0 deletions.
Empty file added tests/__init__.py
View file
Empty file.
180 changes: 180 additions & 0 deletions tests/test_util.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,180 @@
'''
Unit tests for util functions
'''

from datetime import datetime,date
from unittest import TestCase, main
from accounting_reports import util


class TestUtil(TestCase):
'''
Tests for misc. `accounting_reports.util` methods
'''

def test_last_day_of_month_normal(self):
'''
case: normal case
'''
self.assertEqual(util.last_day_of_month(datetime(2017, 1, 1)), datetime(2017, 1, 31))
self.assertEqual(util.last_day_of_month(datetime(2017, 2, 1)), datetime(2017, 2, 28))
self.assertEqual(util.last_day_of_month(datetime(2017, 3, 1)), datetime(2017, 3, 31))
self.assertEqual(util.last_day_of_month(datetime(2017, 4, 1)), datetime(2017, 4, 30))
self.assertEqual(util.last_day_of_month(datetime(2017, 5, 1)), datetime(2017, 5, 31))
self.assertEqual(util.last_day_of_month(datetime(2017, 6, 1)), datetime(2017, 6, 30))
self.assertEqual(util.last_day_of_month(datetime(2017, 7, 1)), datetime(2017, 7, 31))
self.assertEqual(util.last_day_of_month(datetime(2017, 8, 1)), datetime(2017, 8, 31))
self.assertEqual(util.last_day_of_month(datetime(2017, 9, 1)), datetime(2017, 9, 30))
self.assertEqual(util.last_day_of_month(datetime(2017, 10, 1)), datetime(2017, 10, 31))
self.assertEqual(util.last_day_of_month(datetime(2017, 11, 1)), datetime(2017, 11, 30))
self.assertEqual(util.last_day_of_month(datetime(2017, 12, 1)), datetime(2017, 12, 31))


def test_last_day_of_month_leap(self):
'''
case: leap years
'''
self.assertEqual(util.last_day_of_month(datetime(2016, 1, 1)), datetime(2016, 1, 31))
self.assertEqual(util.last_day_of_month(datetime(2016, 2, 1)), datetime(2016, 2, 29))
self.assertEqual(util.last_day_of_month(datetime(2016, 3, 1)), datetime(2016, 3, 31))
self.assertEqual(util.last_day_of_month(datetime(2016, 4, 1)), datetime(2016, 4, 30))
self.assertEqual(util.last_day_of_month(datetime(2016, 5, 1)), datetime(2016, 5, 31))
self.assertEqual(util.last_day_of_month(datetime(2016, 6, 1)), datetime(2016, 6, 30))
self.assertEqual(util.last_day_of_month(datetime(2016, 7, 1)), datetime(2016, 7, 31))
self.assertEqual(util.last_day_of_month(datetime(2016, 8, 1)), datetime(2016, 8, 31))
self.assertEqual(util.last_day_of_month(datetime(2016, 9, 1)), datetime(2016, 9, 30))
self.assertEqual(util.last_day_of_month(datetime(2016, 10, 1)), datetime(2016, 10, 31))
self.assertEqual(util.last_day_of_month(datetime(2016, 11, 1)), datetime(2016, 11, 30))
self.assertEqual(util.last_day_of_month(datetime(2016, 12, 1)), datetime(2016, 12, 31))

'''
case: exceptional leap years
'''
self.assertEqual(util.last_day_of_month(datetime(1900, 2, 1)), datetime(1900, 2, 28))
self.assertEqual(util.last_day_of_month(datetime(2000, 2, 1)), datetime(2000, 2, 29))


def test_end_or_default_empty(self):
'''
case: empty string as parameter
'''
today = date.today()
last_day = util.last_day_of_month(today)
expected = date(today.year, today.month, last_day.day)
actual = util.end_or_default('')
self.assertEqual(expected, actual)


def test_end_or_default_none(self):
'''
case: `None` as parameter
'''
today = date.today()
last_day = util.last_day_of_month(today)
expected = date(today.year, today.month, last_day.day)
actual = util.end_or_default(None)
self.assertEqual(expected, actual)


def test_end_or_default_normal(self):
'''
Normal case
'''
test_case = '2018-06-30'
expected = date(2018, 6, 30)
actual = util.end_or_default(test_case)
self.assertEqual(expected, actual)


def test_begin_or_default_empty(self):
'''
case: empty string as parameter
'''
expected = date(date.today().year, 1, 1)
actual = util.begin_or_default('')
self.assertEqual(expected, actual)


def test_begin_or_default_none(self):
'''
case: `None` as parameter
'''
expected = date(date.today().year, 1, 1)
actual = util.begin_or_default(None)
self.assertEqual(expected, actual)


def test_begin_or_default_normal(self):
'''
Normal case
'''
test_case = '2018-06-30'
expected = date(2018, 6, 30)
actual = util.begin_or_default(test_case)
self.assertEqual(expected, actual)


def test_list_of_months_from_dec(self):
'''
confirms that the list of months returned:
- handles YoY overlap
- handles leap years
- properly includes begin/end dates
'''
begin = datetime(2015, 12, 1)
end = datetime(2017, 3, 1)

actual = util.list_of_months_from(begin, end)
self.assertEqual(len(actual), 16)
self.assertEqual(actual[0], datetime(2015, 12, 31))
self.assertEqual(actual[1], datetime(2016, 1, 31))
self.assertEqual(actual[2], datetime(2016, 2, 29))
self.assertEqual(actual[3], datetime(2016, 3, 31))
self.assertEqual(actual[4], datetime(2016, 4, 30))
self.assertEqual(actual[5], datetime(2016, 5, 31))
self.assertEqual(actual[6], datetime(2016, 6, 30))
self.assertEqual(actual[7], datetime(2016, 7, 31))
self.assertEqual(actual[8], datetime(2016, 8, 31))
self.assertEqual(actual[9], datetime(2016, 9, 30))
self.assertEqual(actual[10], datetime(2016, 10, 31))
self.assertEqual(actual[11], datetime(2016, 11, 30))
self.assertEqual(actual[12], datetime(2016, 12, 31))
self.assertEqual(actual[13], datetime(2017, 1, 31))
self.assertEqual(actual[14], datetime(2017, 2, 28))
self.assertEqual(actual[15], datetime(2017, 3, 31))

def test_csv_to_list_normal(self):
'''
normal case
'''
expected = ['a', 'b', 'c']
actual = util.csv_to_list('a,b,c')
self.assertSequenceEqual(expected, actual)

def test_csv_to_list_empty(self):
'''
empty string case
'''
expected = []
actual = util.csv_to_list('')
self.assertSequenceEqual(expected, actual)

def test_csv_to_list_none(self):
'''
`None` string case
'''
expected = []
actual = util.csv_to_list(None)
self.assertSequenceEqual(expected, actual)

def test_csv_to_list_one_comma(self):
'''
String with one comma case
'''
expected = ['', '']
actual = util.csv_to_list(',')
self.assertSequenceEqual(expected, actual)


if __name__ == '__main__':
main()

0 comments on commit c0d0eab

Please sign in to comment.