From 4b28beb600e6f8e39ea2eb9af0e870fe43eeb9c5 Mon Sep 17 00:00:00 2001 From: "A. Junior" Date: Fri, 2 May 2025 01:00:35 -0300 Subject: [PATCH] feat: add rows count --- table.go | 5 +++++ table_test.go | 10 ++++++++++ 2 files changed, 15 insertions(+) diff --git a/table.go b/table.go index 6daaebf..cbaceb9 100644 --- a/table.go +++ b/table.go @@ -939,3 +939,8 @@ func (t *Table) Render() { func (t *Table) IsEmpty() bool { return len(t.data) == 0 } + +// RowCount returns the number of rows in the table. +func (t *Table) RowCount() int { + return len(t.data) +} diff --git a/table_test.go b/table_test.go index cbfd55e..0a027bb 100644 --- a/table_test.go +++ b/table_test.go @@ -830,3 +830,13 @@ func Test_TableIsNotEmpty(t *testing.T) { table.AddRow("4", "5", "6") assert.Equal(t, false, table.IsEmpty()) } + +func Test_TableRowCount(t *testing.T) { + builder := &strings.Builder{} + table := New(builder) + table.SetHeaders("A", "B", "C") + assert.Equal(t, 0, table.RowCount()) + table.AddRow("1", "2", "3") + table.AddRow("4", "5", "6") + assert.Equal(t, 2, table.RowCount()) +}