def test_group_period_extract_both(self):
first = [ "01/01/2014", "-1.00", "Foo" ]
second = [ "02/01/2014", "-2.00", "Foo" ]
transactions = [ first, second ]
expected = [ { "01/2014" : [ first, second ] }, { "2014:00" : [ first, second ] } ]
result = visualise.group_period(transactions, [ visualise.extract_month, visualise.extract_week ])
self.assertEquals(2, len(result))
self.assertEquals(expected, result)
def test_group_period_extract_month(self):
first = [ "01/01/2014", "-1.00", "Foo" ]
second = [ "09/01/2014", "-2.00", "Bar" ]
month = [ first, second ]
expected = { "01/2014" : month }
result = visualise.group_period(month, [ visualise.extract_month ])
self.assertEquals(1, len(result))
self.assertEquals(expected, result[0])
def test_group_period_extract_week(self):
first = [ "01/01/2014", "-1.00", "Foo" ]
second = [ "09/01/2014", "-2.00", "Bar" ]
week = [ first, second ]
expected = { "2014:00" : [ first ], "2014:01" : [ second ] }
result = visualise.group_period(week, [ visualise.extract_week ])
self.assertEquals(1, len(result))
self.assertEquals(expected, result[0])
def test_group_period_empty(self):
self.assertEquals({}, visualise.group_period([])[0])
