def test_table_length():
sheet = dtocean_dummy.Spreadsheet()
array = sheet._get_random_array(10)
sheet._array_2_df(array)
assert len(sheet.table) == 10
def test_table_cumsum_total():
'''Check that the cumulative sum in the pandas table is OK'''
sheet = dtocean_dummy.Spreadsheet()
array = sheet._get_random_array(10)
sheet._array_2_df(array)
sheet._add_cumsum()
cumsum = sheet.table['Cumulative']
assert cumsum.irow(-1) == sum(array)
def test_table_cumsum_length():
'''Check that the length of the cumulative sum is OK'''
sheet = dtocean_dummy.Spreadsheet()
array = sheet._get_random_array(10)
sheet._array_2_df(array)
sheet._add_cumsum()
cumsum = sheet.table['Cumulative']
assert len(cumsum) == 10
def test_create_xls(tmpdir):
'''Test if excel file is written'''
locd = tmpdir.mkdir("sub")
p = locd.join("test.xls")
sheet = dtocean_dummy.Spreadsheet()
sheet.make_table(10)
sheet.write_xls(str(p))
assert len(locd.listdir()) == 1
assert os.path.basename(str(locd.listdir()[0])) == "test.xls"
def test_spreadsheet_call(tmpdir):
'''Test if Spreadsheet.call works for all file types'''
locd = tmpdir.mkdir("sub")
dirp = str(locd)
sheet = dtocean_dummy.Spreadsheet()
# Check the error for a bad file type
filep = os.path.join(dirp, "test.docx")
with pytest.raises(ValueError):
sheet(10, filep, out_fmt='docx')
for ftype in sheet._valid_formats:
filep = os.path.join(dirp, "test.{}".format(ftype))
sheet(10, filep, out_fmt=ftype)
assert len(locd.listdir()) == len(sheet._valid_formats)
def sheet():
'''Share a Spreadsheet object'''
return dtocean_dummy.Spreadsheet()