def test_income_only_two_months(self):
month = [ "01/2014", "02/2014" ]
amount = 1.00
data = { month[0] : { "Income" : amount },
month[1] : { "Income" : amount } }
expected = { month[0] : amount, month[1] : amount }
self.assertEquals(expected, visualise.income(data))
def test_income_mix(self):
month = "01/2014"
amount = 1.00
data = { month : { visualise.whitelist[0] : "0.00", "Income" : amount } }
expected = { month : amount }
self.assertEquals(expected, visualise.income(data))