def test_get_header(self, field_names_to_extract): field_names_to_extract.return_value = [ "episode_id", "name", "consistency_token", "id" ] renderer = extract.EpisodeSubrecordCsvRenderer( Colour, models.Episode.objects.all(), self.user) self.assertEqual(["Patient", "Episode", "Name"], renderer.get_headers())
def test_get_row(self, field_names_to_extract): field_names_to_extract.return_value = [ "episode_id", "name", "consistency_token", "id" ] renderer = extract.EpisodeSubrecordCsvRenderer( Colour, models.Episode.objects.all(), self.user) rendered = renderer.get_row(self.colour) self.assertEqual(["1", "1", "blue"], rendered)