def test_add_get_methods(self):
table = OutputTable(headers=["Column 0"])
# add/get single row/col
table.add_row(["value 0,0"])
self.assertEqual(table.get_row(index=0), ["value 0,0"])
table.add_column(["Column 1", "value 0,1"])
self.assertEqual(table.headers(), ["Column 0", "Column 1"])
self.assertEqual(table.get_row(index=0), ["value 0,0", "value 0,1"])
self.assertEqual(table.get_column(index=1), ["Column 1", "value 0,1"])
# add/get row and column by index
table.add_row(["value -1,0", "value -1,1"], index=0)
self.assertEqual(table.get_row(index=0), ["value -1,0", "value -1,1"])
self.assertEqual(table.get_column(index=0),
["Column 0", "value -1,0", "value 0,0"])
self.assertEqual(table.get_column(index=1),
["Column 1", "value -1,1", "value 0,1"])
table.add_column(["Column 0.5", "value -1,0.5", "value 0,0.5"],
index=1)
self.assertEqual(table.get_column(index=1),
["Column 0.5", "value -1,0.5", "value 0,0.5"])
self.assertEqual(table.get_row(index=0),
["value -1,0", "value -1,0.5", "value -1,1"])
self.assertEqual(table.get_row(index=1),
["value 0,0", "value 0,0.5", "value 0,1"])
def test_create_headers(self):
table = OutputTable(headers=["Column 0"])
self.assertEqual(table.headers(), ["Column 0"])
def test_create_headers(self):
table = OutputTable(headers=["Column 0"])
self.assertEqual(table.headers(), ["Column 0"])
self.assertEqual(table.column_widths(),
[len("Column 0") + OutputTable.column_padding])