def test_ignore_several(self): month = "01/2014" amount = -1.00 ignore = [ "Cash", "Dining" ] data = { month : dict((i, amount) for i in ignore) } expected = { month : {} } self.assertEquals(expected, visualise.ignore(data, *ignore))
def test_ignore_one(self): month = "01/2014" amount = -1.00 single = "Cash" data = { month : { single : amount } } expected = { month : {} } self.assertEquals(expected, visualise.ignore(data, single))