improve test coverage

This commit is contained in:
chfw 2016-12-16 22:29:44 +00:00
parent e8fa729c46
commit 4fff82d8cf
3 changed files with 24 additions and 0 deletions

View File

@ -20,6 +20,12 @@ def test_book_reader():
reader.read_sheet(1)
@raises(IOError)
def test_book_reader_open_stream():
reader = BookReader()
reader.open_stream("a string")
@raises(IOError)
def test_book_writer():
writer = BookWriter()

View File

@ -113,6 +113,15 @@ def test_get_io():
assert io is None
def test_get_io_type():
t = manager.get_io_type("hello")
assert t is None
t = manager.get_io_type("csv")
eq_(t, 'string')
t = manager.get_io_type("xls")
eq_(t, 'bytes')
def test_default_csv_format():
data = [['1', '2', '3']]
io = manager.get_io("csv")
@ -189,6 +198,13 @@ def test_library_parameter():
assert result['csv'] == [[1, 2, 3]]
@raises(Exception)
def test_library_parameter_error_situation():
data = [['1', '2', '3']]
io = manager.get_io("csv")
save_data(io, data, 'csv', library="doesnot-exist")
def test_conversion_from_bytes_to_text():
test_file = "conversion.csv"
data = [['1', '2', '3']]

View File

@ -13,3 +13,5 @@ def test_write_empty_array():
writer = MyWriter(test_string, "somesheet", "somename")
writer.write_array([])
eq_(test_string, writer._native_book)