def assertEmptyQuerySetMatches(self, expected_data, **kwargs): qs = self.qs.none() obj = StringIO() if DJANGO_VERSION[:2] == (1, 5): with self.assertRaises(djqscsv.CSVException): djqscsv.write_csv(qs, obj) else: djqscsv.write_csv(qs, obj, **kwargs) self.assertEqual(obj.getvalue(), expected_data)
def assertEmptyQuerySetMatches(self, expected_data, **kwargs): qs = self.qs.none() obj = StringIO() if DJANGO_VERSION[:2] == (1, 5): with self.assertRaises(djqscsv.CSVException): djqscsv.write_csv(qs, obj) elif DJANGO_VERSION[:2] == (1, 6): djqscsv.write_csv(qs, obj, **kwargs) self.assertEqual(obj.getvalue(), expected_data)
def assertQuerySetBecomesCsv(self, qs, expected_data, **kwargs): obj = BytesIO() djqscsv.write_csv(qs, obj, **kwargs) csv_file = filter(None, obj.getvalue().splitlines()) self.assertMatchesCsv(csv_file, expected_data)
def assertQuerySetBecomesCsv(self, qs, expected_data, **kwargs): obj = StringIO() djqscsv.write_csv(qs, obj, **kwargs) csv_file = filter(None, obj.getvalue().split('\n')) self.assertMatchesCsv(csv_file, expected_data)