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())