1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
# -*- coding: utf-8 -*-
"""
Unit tests for the CCSV write module.
Author: Gertjan van den Burg
"""
import csv
import tempfile
import unittest
import clevercsv
from clevercsv.dialect import SimpleDialect
class WriterTestCase(unittest.TestCase):
def writerAssertEqual(self, input, expected_result):
with tempfile.TemporaryFile("w+", newline="", prefix="ccsv_") as fp:
writer = clevercsv.writer(fp, dialect=self.dialect)
writer.writerows(input)
fp.seek(0)
self.assertEqual(fp.read(), expected_result)
def _write_test(self, fields, expect, **kwargs):
with tempfile.TemporaryFile("w+", newline="", prefix="ccsv_") as fp:
writer = clevercsv.writer(fp, **kwargs)
writer.writerow(fields)
fp.seek(0)
self.assertEqual(fp.read(), expect + writer.dialect.lineterminator)
def _write_error_test(self, exc, fields, **kwargs):
with tempfile.TemporaryFile("w+", newline="", prefix="ccsv_") as fp:
writer = clevercsv.writer(fp, **kwargs)
with self.assertRaises(exc):
writer.writerow(fields)
fp.seek(0)
self.assertEqual(fp.read(), "")
def test_write_arg_valid(self):
self._write_error_test(clevercsv.Error, None)
self._write_test((), "")
self._write_test([None], '""')
self._write_error_test(
clevercsv.Error, [None], quoting=clevercsv.QUOTE_NONE
)
# Check that exceptions are passed up the chain
class BadList:
def __len__(self):
return 10
def __getitem__(self, i):
if i > 2:
raise OSError
self._write_error_test(OSError, BadList())
class BadItem:
def __str__(self):
raise OSError
self._write_error_test(OSError, [BadItem()])
def test_write_bigfield(self):
bigstring = "X" * 50000
self._write_test(
[bigstring, bigstring], "%s,%s" % (bigstring, bigstring)
)
def test_write_quoting(self):
self._write_test(["a", 1, "p,q"], 'a,1,"p,q"')
self._write_error_test(
clevercsv.Error, ["a", 1, "p,q"], quoting=clevercsv.QUOTE_NONE
)
self._write_test(
["a", 1, "p,q"], 'a,1,"p,q"', quoting=clevercsv.QUOTE_MINIMAL
)
self._write_test(
["a", 1, "p,q"], '"a",1,"p,q"', quoting=clevercsv.QUOTE_NONNUMERIC
)
self._write_test(
["a", 1, "p,q"], '"a","1","p,q"', quoting=clevercsv.QUOTE_ALL
)
self._write_test(
["a\nb", 1], '"a\nb","1"', quoting=clevercsv.QUOTE_ALL
)
def test_write_simpledialect(self):
self._write_test(
["a", 1, "p,q"],
"a,1,|p,q|",
dialect=SimpleDialect(delimiter=",", quotechar="|", escapechar=""),
)
def test_write_csv_dialect(self):
self._write_test(
["a", 1, "p,q"],
'a,1,"p,q"',
dialect="excel",
)
self._write_test(
["a", 1, "p,q"],
'"a","1","p,q"',
dialect=csv.unix_dialect,
)
self._write_test(
[1, 2, 3],
"1\t2\t3",
dialect=clevercsv.excel_tab,
)
if __name__ == "__main__":
unittest.main()
|
