Use table style mappings during document conversion

1 files changed, 3 insertions, 2 deletions

diff --git a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
index 0289bdb..f206640 100644
--- a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
+++ b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
@@ -122,8 +122,9 @@ public class DocumentElementMakers {
         return run(withChildren(new Text(text)));
     }
 
-    public static Table table(List<DocumentElement> rows) {
-        return new Table(Optional.empty(), rows);
+    public static Table table(List<DocumentElement> rows, Object... args) {
+        Arguments arguments = new Arguments(args);
+        return new Table(arguments.get(STYLE, Optional.empty()), rows);
     }
 
     public static TableRow tableRow(List<DocumentElement> cells, Object... args) {