def test_create_csv(self):
writer = CSVWriter()
writer.add_row([1, 'a'])
writer.add_row([2, 'b'])
csv = writer.get_csv()
self.assertIsInstance(csv, str)
self.assertEqual(csv, b'1,"a"\n2,"b"\n')
def write_prev_prop_pub(file_, prev_proposals):
"""
Write a CSV file of previous proposal publication information.
"""
writer = CSVWriter()
writer.add_row(
[
"Submitted proposal",
"Referenced proposal",
"Publication type",
"Publication description",
"Publication author",
"Publication year",
"Publication title",
]
)
for (this_proposal_code, proposals) in prev_proposals.items():
for proposal in proposals.values():
for publication in proposal.publications:
writer.add_row(
[
this_proposal_code,
proposal.proposal_code,
PublicationType.get_name(publication.type),
publication.description,
publication.author,
publication.year,
publication.title,
]
)
file_.write(writer.get_csv())
