class TestDonationsParser(unittest.TestCase): """Testing DonationsParser class.""" def setUp(self): self.donations_parser = DonationsParser(2014, 01) donations_data = { '01': {'sum': 370, 'avg': 46.25, 'quantity': 8}, '02': {'sum': 5682, 'avg': 132.14, 'quantity': 43} } self.donations_parser.donations = donations_data def test_get_csv(self): expected = """'day', 'sum', 'quantity', 'avg' '2014-01-01', 370, 8, 46.25 '2014-01-02', 5682, 43, 132.14 """ output = self.donations_parser.get_csv() self.assertEqual(output, expected) def test_get_js(self): expected = """var my_var = [ ['day', 'sum', 'quantity', 'avg'], ['2014-01-01', 370, 8, 46.25], ['2014-01-02', 5682, 43, 132.14] ];""" output = self.donations_parser.get_js('my_var') self.assertEqual(output, expected)
def setUp(self): self.donations_parser = DonationsParser(2014, 01) donations_data = { '01': {'sum': 370, 'avg': 46.25, 'quantity': 8}, '02': {'sum': 5682, 'avg': 132.14, 'quantity': 43} } self.donations_parser.donations = donations_data