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)