From e4f8c5bad42b36712c50ef94f265e8e4e10a4d53 Mon Sep 17 00:00:00 2001
From: Michal Jaron <mj@optimy.pl>
Date: Tue, 7 Sep 2021 11:32:41 +0200
Subject: [PATCH] Unit test case fix.

---
 .../java/pl/mjaron/etudes/flat/TableColumnsWidthComputer.java | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main/java/pl/mjaron/etudes/flat/TableColumnsWidthComputer.java b/src/main/java/pl/mjaron/etudes/flat/TableColumnsWidthComputer.java
index 30022cb..d794690 100644
--- a/src/main/java/pl/mjaron/etudes/flat/TableColumnsWidthComputer.java
+++ b/src/main/java/pl/mjaron/etudes/flat/TableColumnsWidthComputer.java
@@ -33,7 +33,9 @@ private static void apply(final int[] widths, final Iterable<String> series) {
 
     public static int[] compute(final ITableSource source) {
         final int[] widths = new int[source.getColumnsCount()];
-        apply(widths, source.getHeaders());
+        if (source.hasHeaders()) {
+            apply(widths, source.getHeaders());
+        }
         for (final Iterable<String> row : source) {
             apply(widths, row);
         }