def run_task(self, source):
"""Helper utility for running task under test"""
def reformat(string):
"""Reformat string to make it like a TSV."""
return textwrap.dedent(string).strip().replace(' ', '\t')
with open(self.input_file, 'w') as fle:
fle.write(reformat(source))
task = TotalEventsReport(counts=self.input_file,
report='fake_report')
output_target = FakeTarget()
task.output = MagicMock(return_value=output_target)
task.run()
data = output_target.buffer.read()
result = pandas.read_csv(StringIO(data),
na_values=['-'],
index_col=False,
header=None,
names=['date', 'count'])
return result
def test_create_csv_entry(self):
entry = TotalEventsReport.create_csv_entry('2014-01-02', '5000')
self.assertEqual(entry, '2014-01-02,5000')
def test_create_csv_header(self):
header = TotalEventsReport.create_header()
self.assertEqual(header, "date,event_count")
