def test_export_xlsx_temp_del2(self):
'''Verify write xslx, named tempfile; delete afterwards.'''
# Write named tempfiles, verify, then remove
workbook_fpath, = ut.export(
self.LM, overwrite=True, suffix='.xlsx', temp=True, keepname=True,
delete=True
)
actual1 = os.path.exists(workbook_fpath)
nt.assert_false(actual1)
def test_export_csv_temp_del2(self):
'''Verify write csv, named tempfile; delete afterwards.'''
# Write named tempfiles, verify, then remove
data_fpath, dash_fpath = ut.export(
self.LM, overwrite=True, suffix='.csv', temp=True, keepname=True,
delete=True
)
actual1 = os.path.exists(data_fpath)
actual2 = os.path.exists(dash_fpath)
nt.assert_false(actual1)
nt.assert_false(actual2)
def test_export_xlsx_temp1(self):
'''Verify write xslx files, unnamed tempfile.'''
try:
# Write unnamed tempfiles and see if exists; then remove.
workbook_fpath, = ut.export(
self.LM, overwrite=True, suffix='.xlsx', temp=True, keepname=False,
delete=False
)
actual1 = os.path.exists(workbook_fpath)
nt.assert_true(actual1)
finally:
os.remove(workbook_fpath)
logging.info('File has been deleted: {}'.format(workbook_fpath))
def test_export_xlsx_temp2(self):
'''Verify write xslx files; named tempfile.'''
try:
# Write named tempfiles, verify, then remove
workbook_fpath, = ut.export(
self.LM, overwrite=True, suffix='.xlsx', temp=True, keepname=True,
delete=False
)
actual1 = workbook_fpath
expected1 = self.xlsx_fpath
nt.assert_equals(actual1, expected1)
finally:
os.remove(self.xlsx_fpath)
logging.info('File has been deleted: {}'.format(workbook_fpath))
def to_xlsx(self, offset=3, **kwargs):
'''Write LaminateModel data FeatureInput dashboard as one file.
Returns
-------
tuple
Path for .xlsx file; maintained For type consistency.
See Also
--------
- `utils.tools.export`: for kwargs and docstring.
'''
(workbook_fpath,) = ut.export(self, suffix='.xlsx', offset=offset, **kwargs)
return (workbook_fpath,)
def to_csv(self, **kwargs):
'''Write LaminateModel data FeatureInput dashboard as separate files.
Returns
-------
tuple
Paths for both .csv files.
See Also
--------
- `utils.tools.export`: for kwargs and docstring.
'''
data_fpath, dash_fpath = ut.export(self, suffix='.csv', **kwargs)
return data_fpath, dash_fpath
def test_export_csv_temp2(self):
'''Verify write csv files; named tempfile.'''
try:
# Write named tempfiles, verify, then remove
data_fpath, dash_fpath = ut.export(
self.LM, overwrite=True, suffix='.csv', temp=True, keepname=True,
delete=False
)
actual1, actual2 = data_fpath, dash_fpath
expected1, expected2 = self.csv_fpath, self.csv_dash_fpath
nt.assert_equals(actual1, expected1)
nt.assert_equals(actual2, expected2)
finally:
os.remove(data_fpath)
os.remove(dash_fpath)
logging.info('File has been deleted: {}'.format(data_fpath))
logging.info('File has been deleted: {}'.format(dash_fpath))
def test_export_csv_temp1(self):
'''Verify write csv files, unnamed tempfile.'''
try:
# Write unnamed tempfiles and see if exists; then remove.
data_fpath, dash_fpath = ut.export(
self.LM, overwrite=True, suffix='.csv', temp=True, keepname=False,
delete=False
)
actual1 = os.path.exists(data_fpath)
actual2 = os.path.exists(dash_fpath)
nt.assert_true(actual1)
nt.assert_true(actual2)
finally:
os.remove(data_fpath)
os.remove(dash_fpath)
logging.info('File has been deleted: {}'.format(data_fpath))
logging.info('File has been deleted: {}'.format(dash_fpath))
def test_export_csv_temp_increment1(self):
'''Verify writes incremented csv if same filename found; named tempfile.
Notes
-----
Test the data file, but since dashboards are written, deletes both files.
Simply sees if incremented file is greater. If incremented file is empty
(starting fresh), then ignore loop.
'''
fname_list = []
try:
baseline = self.csv_fpath
for i in range(3):
# Increment filepath after first loop
data_fpath, dash_fpath = ut.export(
self.LM, overwrite=False, prefix=None, suffix='.csv',
temp=True, keepname=True, delete=False
)
# See differences (should be the increment); assert last number < current
incremented = data_fpath
prior_inc = [int(i[2:]) for i in difflib.ndiff(baseline, incremented)
if '-' in i and i[2:].isdigit()]
post_inc = [int(i[2:]) for i in difflib.ndiff(baseline, incremented)
if '+' in i and i[2:].isdigit()]
if prior_inc == []: prior_inc = [0] # reset prior
if post_inc: # ignore loop if [], means first loop, and hasn't incremented yet
actual = prior_inc[0] < post_inc[0]
nt.assert_true(actual)
baseline = data_fpath
fname_list.append(data_fpath)
fname_list.append(dash_fpath) # not used by still written, so need cleaning
finally:
for fpath in fname_list:
os.remove(fpath)
logging.info('File has been deleted: {}'.format(fpath))
def test_export_xlsx_error1(self):
'''Verify raises error if a directory name is passed in.'''
# This way for now due to security concerns accidentatly writing to disk.
ut.export(self.LM, dirpath='dummy', delete=True)
def test_export_xlsx_default(self):
'''Verify default suffix is '.xlsx'.'''
result = ut.export(self.LM, suffix=None, temp=True, delete=True)
actual = result[0].endswith('.xlsx')
nt.assert_true(actual)
