def test_small_set(self): """Tests a small data implementation""" listdata = [('Agatha', 'Monday, March 16, 2015 05:00PM'), ('Arthur', 'Monday, March 16, 2015 05:20PM')] expected = [self.fmtstr.format(*v) for v in listdata] returned = task_02.prepare_email(listdata) self.assertEqual(returned, expected)
def test_large_set(self): """Tests a large set of data""" listdata = data.get_email_data(1000) expected = [self.fmtstr.format(*v) for v in listdata] returned = task_02.prepare_email(listdata) self.assertEqual(returned, expected)