#302: adds unit test and doc tidying.

This commit is contained in:
Romain Dorgueil
2019-06-01 09:32:22 +02:00
parent 94c5f1a45a
commit dee719f070
2 changed files with 19 additions and 16 deletions

View File

@ -89,6 +89,22 @@ class CsvWriterTest(Csv, WriterTest, TestCase):
assert self.readlines() == ("foo,bar", "a,b", "c,d")
@incontext(skip_header=False)
def test_fields_with_headers(self, context):
context.set_input_fields(["foo", "bar"])
context.write_sync(("a", "b"), ("c", "d"))
context.stop()
assert self.readlines() == ("foo,bar", "a,b", "c,d")
@incontext(skip_header=True)
def test_fields_without_headers(self, context):
context.set_input_fields(["foo", "bar"])
context.write_sync(("a", "b"), ("c", "d"))
context.stop()
assert self.readlines() == ("a,b", "c,d")
@incontext()
def test_fields_from_type(self, context):
context.set_input_type(namedtuple("Point", "x y"))