def test_write_data_to_file(self): """Write data to file """ tabfile = TabFile(fp=self.fp) out_file = os.path.join(self.working_dir, "test.tsv") tabfile.write(filen=out_file) self.assertEqual(io.open(out_file, 'rt').read(), self.data)
def test_write_data(self): """Write data to file-like object """ tabfile = TabFile('test',self.fp) fp = cStringIO.StringIO() tabfile.write(fp=fp) self.assertEqual(fp.getvalue(),self.data) fp.close()
def test_write_data_with_header(self): """Write data to file-like object including a header line """ tabfile = TabFile('test',self.fp,first_line_is_header=True,delimiter=',') fp = cStringIO.StringIO() tabfile.write(fp=fp,include_header=True) self.assertEqual(fp.getvalue(),self.header.replace('\t',',')+self.data.replace('\t',',')) fp.close()
def test_write_data(self): """Write data to file-like object """ tabfile = TabFile('test', self.fp) fp = cStringIO.StringIO() tabfile.write(fp=fp) self.assertEqual(fp.getvalue(), self.data) fp.close()
def test_write_data_to_file_include_header(self): """Write data to file including header """ tabfile = TabFile(fp=self.fp, first_line_is_header=True) out_file = os.path.join(self.working_dir, "test.tsv") tabfile.write(filen=out_file, include_header=True) self.assertEqual( io.open(out_file, 'rt').read(), self.header + self.data)
def test_write_data_include_header(self): """Write data to file-like object including header """ tabfile = TabFile(fp=self.fp, first_line_is_header=True) fp = io.StringIO() tabfile.write(fp=fp, include_header=True) self.assertEqual(fp.getvalue(), self.header + self.data) fp.close()
def test_write_data_with_header(self): """Write data to file-like object including a header line """ tabfile = TabFile('test', self.fp, first_line_is_header=True, delimiter=',') fp = cStringIO.StringIO() tabfile.write(fp=fp, include_header=True) self.assertEqual( fp.getvalue(), self.header.replace('\t', ',') + self.data.replace('\t', ',')) fp.close()
def test_change_delimiter_for_write(self): """Write data out with different delimiter to input """ tabfile = TabFile('test',self.fp,delimiter=',') # Modified delimiter (tab) fp = cStringIO.StringIO() tabfile.write(fp=fp,delimiter='\t') self.assertEqual(fp.getvalue(),self.data) fp.close() # Default (should revert to comma) fp = cStringIO.StringIO() tabfile.write(fp=fp) self.assertEqual(fp.getvalue(),self.data.replace('\t',',')) fp.close()
def test_change_delimiter_for_write(self): """Write data out with different delimiter to input """ tabfile = TabFile('test', self.fp, delimiter=',') # Modified delimiter (tab) fp = cStringIO.StringIO() tabfile.write(fp=fp, delimiter='\t') self.assertEqual(fp.getvalue(), self.data) fp.close() # Default (should revert to comma) fp = cStringIO.StringIO() tabfile.write(fp=fp) self.assertEqual(fp.getvalue(), self.data.replace('\t', ',')) fp.close()