diff --git a/table/table.go b/table/table.go index 4cd1329..e4f47bc 100644 --- a/table/table.go +++ b/table/table.go @@ -339,7 +339,8 @@ func rowToString(rows []Row) [][]string { } // SetHeaders sets the table headers. -func (m *Model) SetHeaders(headers []string) { +// TODO should this be variadic to match lipgloss table? +func (m *Model) SetHeaders(headers ...string) { m.headers = headers m.table.Headers(headers...) } diff --git a/table/table_test.go b/table/table_test.go index 08893b7..051bb7c 100644 --- a/table/table_test.go +++ b/table/table_test.go @@ -13,7 +13,7 @@ func TestFromValues(t *testing.T) { t.Run("Headers", func(t *testing.T) { input := "foo1,bar1\nfoo2,bar2\nfoo3,bar3" table := New() - table.Headers("Foo", "Bar") + table.SetHeaders("Foo", "Bar") table.FromValues(input, ",") if len(table.rows) != 3 {