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'], '+')