def test_set_header(self):
# Rename columns
tbl = Table([['one', 'two'], [1, 2], [3, 4]])
new_tbl = tbl.set_header(['oneone', 'twotwo'])
self.assertEqual(new_tbl[0], {'oneone': 1, 'twotwo': 2})
# Change number of columns
tbl = Table([['one', 'two'], [1, 2], [3, 4]])
new_tbl = tbl.set_header(['one'])
self.assertEqual(new_tbl[0], {'one': 1})
