def get_dates(transactions): dates = rt.all_dates(transactions) dates.append(date.today()) dates.sort() return dates
def test_all_dates(): t = rt.read_file('transactions') l = rt.all_dates(t) assert l.count(date(2014, 1, 10)) == 1 assert l.count(date(2014, 2, 10)) == 1 assert l.count(date(2014, 4, 10)) == 1