Unit test case fix.

mjfryc · Sep 7, 2021 · e4f8c5b · e4f8c5b
1 parent bb0ad2f
commit e4f8c5b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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);
         }