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()
