Commit 75f882c5 authored by Andrew McNamara's avatar Andrew McNamara

Improve test coverage fractionally.

parent 1b58427c
......@@ -30,6 +30,8 @@ class Test_Csv(unittest.TestCase):
self.assertRaises(TypeError, ctor, arg, lineterminator=None)
self.assertRaises(TypeError, ctor, arg, lineterminator=1)
self.assertRaises(TypeError, ctor, arg, quoting=None)
self.assertRaises(TypeError, ctor, arg,
quoting=csv.QUOTE_ALL, quotechar='')
def test_reader_arg_valid(self):
self._test_arg_valid(csv.reader, [])
......@@ -231,6 +233,8 @@ class Test_Csv(unittest.TestCase):
def test_read_bigfield(self):
# This exercises the buffer realloc functionality and field size
# limits.
limit = csv.set_field_limit()
try:
size = 50000
bigstring = 'X' * size
bigline = '%s,%s' % (bigstring, bigstring)
......@@ -240,6 +244,10 @@ class Test_Csv(unittest.TestCase):
self.assertEqual(csv.set_field_limit(), size)
csv.set_field_limit(size-1)
self.assertRaises(csv.Error, self._read_test, [bigline], [])
self.assertRaises(TypeError, csv.set_field_limit, None)
self.assertRaises(TypeError, csv.set_field_limit, 1, None)
finally:
csv.set_field_limit(limit)
class TestDialectRegistry(unittest.TestCase):
def test_registry_badargs(self):
......@@ -252,6 +260,12 @@ class TestDialectRegistry(unittest.TestCase):
self.assertRaises(csv.Error, csv.unregister_dialect, "nonesuch")
self.assertRaises(TypeError, csv.register_dialect, None)
self.assertRaises(TypeError, csv.register_dialect, None, None)
self.assertRaises(TypeError, csv.register_dialect, "nonesuch", 0, 0)
self.assertRaises(TypeError, csv.register_dialect, "nonesuch",
badargument=None)
self.assertRaises(TypeError, csv.register_dialect, "nonesuch",
quoting=None)
self.assertRaises(TypeError, csv.register_dialect, [])
def test_registry(self):
class myexceltsv(csv.excel):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment