def test_append_column(self):
"""Append new column to a Tabfile
"""
tabfile = TabFile('test',self.fp,first_line_is_header=True)
self.assertEqual(len(tabfile.header()),4)
tabfile.appendColumn('new')
self.assertEqual(len(tabfile.header()),5)
self.assertEqual(tabfile.header()[4],'new')
self.assertEqual(tabfile[0]['new'],'')
def test_append_column(self):
"""Append new column to a Tabfile
"""
tabfile = TabFile('test', self.fp, first_line_is_header=True)
self.assertEqual(len(tabfile.header()), 4)
tabfile.appendColumn('new')
self.assertEqual(len(tabfile.header()), 5)
self.assertEqual(tabfile.header()[4], 'new')
self.assertEqual(tabfile[0]['new'], '')
def test_set_column_to_constant_value(self):
"""Set a column to a constant value using transformColumn
"""
tabfile = TabFile('test',self.fp,first_line_is_header=True)
# Check number of columns and header items
self.assertEqual(tabfile.nColumns(),4)
self.assertEqual(tabfile.header(),['chr','start','end','data'])
# Add a strand column
tabfile.appendColumn('strand')
self.assertEqual(tabfile.nColumns(),5)
self.assertEqual(tabfile.header(),['chr','start','end','data','strand'])
# Set all values to '+'
tabfile.transformColumn('strand',lambda x: '+')
for line in tabfile:
self.assertEqual(line['strand'],'+')
def test_set_column_to_constant_value(self):
"""Set a column to a constant value using transformColumn
"""
tabfile = TabFile('test', self.fp, first_line_is_header=True)
# Check number of columns and header items
self.assertEqual(tabfile.nColumns(), 4)
self.assertEqual(tabfile.header(), ['chr', 'start', 'end', 'data'])
# Add a strand column
tabfile.appendColumn('strand')
self.assertEqual(tabfile.nColumns(), 5)
self.assertEqual(tabfile.header(),
['chr', 'start', 'end', 'data', 'strand'])
# Set all values to '+'
tabfile.transformColumn('strand', lambda x: '+')
for line in tabfile:
self.assertEqual(line['strand'], '+')
