test(Compare and Comply): Update tests

Author: lpatino10

compare-comply/src/test/java/com/ibm/watson/compare_comply/v1/CompareComplyTest.java

@@ -636,10 +636,11 @@ public void testClassifyElements() throws FileNotFoundException, InterruptedExce
     assertEquals(BEGIN, response.getTables().get(0).getKeyValuePairs().get(0).getKey().getLocation().getBegin());
     assertEquals(END, response.getTables().get(0).getKeyValuePairs().get(0).getKey().getLocation().getEnd());
     assertEquals(TEXT, response.getTables().get(0).getKeyValuePairs().get(0).getKey().getText());
-    assertEquals(CELL_ID, response.getTables().get(0).getKeyValuePairs().get(0).getValue().getCellId());
-    assertEquals(BEGIN, response.getTables().get(0).getKeyValuePairs().get(0).getValue().getLocation().getBegin());
-    assertEquals(END, response.getTables().get(0).getKeyValuePairs().get(0).getValue().getLocation().getEnd());
-    assertEquals(TEXT, response.getTables().get(0).getKeyValuePairs().get(0).getValue().getText());
+    assertEquals(CELL_ID, response.getTables().get(0).getKeyValuePairs().get(0).getValue().get(0).getCellId());
+    assertEquals(BEGIN, response.getTables().get(0).getKeyValuePairs().get(0).getValue().get(0).getLocation()
+        .getBegin());
+    assertEquals(END, response.getTables().get(0).getKeyValuePairs().get(0).getValue().get(0).getLocation().getEnd());
+    assertEquals(TEXT, response.getTables().get(0).getKeyValuePairs().get(0).getValue().get(0).getText());
     assertEquals(TEXT, response.getTables().get(0).getTitle().getText());
     assertEquals(BEGIN, response.getTables().get(0).getTitle().getLocation().getBegin());
     assertEquals(END, response.getTables().get(0).getTitle().getLocation().getEnd());
@@ -795,10 +796,11 @@ public void testExtractTables() throws FileNotFoundException, InterruptedExcepti
     assertEquals(BEGIN, response.getTables().get(0).getKeyValuePairs().get(0).getKey().getLocation().getBegin());
     assertEquals(END, response.getTables().get(0).getKeyValuePairs().get(0).getKey().getLocation().getEnd());
     assertEquals(TEXT, response.getTables().get(0).getKeyValuePairs().get(0).getKey().getText());
-    assertEquals(CELL_ID, response.getTables().get(0).getKeyValuePairs().get(0).getValue().getCellId());
-    assertEquals(BEGIN, response.getTables().get(0).getKeyValuePairs().get(0).getValue().getLocation().getBegin());
-    assertEquals(END, response.getTables().get(0).getKeyValuePairs().get(0).getValue().getLocation().getEnd());
-    assertEquals(TEXT, response.getTables().get(0).getKeyValuePairs().get(0).getValue().getText());
+    assertEquals(CELL_ID, response.getTables().get(0).getKeyValuePairs().get(0).getValue().get(0).getCellId());
+    assertEquals(BEGIN, response.getTables().get(0).getKeyValuePairs().get(0).getValue().get(0).getLocation()
+        .getBegin());
+    assertEquals(END, response.getTables().get(0).getKeyValuePairs().get(0).getValue().get(0).getLocation().getEnd());
+    assertEquals(TEXT, response.getTables().get(0).getKeyValuePairs().get(0).getValue().get(0).getText());
   }
 
   @Test

compare-comply/src/test/resources/compare_comply/classify-return.json

@@ -166,14 +166,16 @@
             },
             "text": "text"
           },
-          "value": {
-            "cell_id": "cell_id",
-            "location": {
-              "begin": 0,
-              "end": 1
-            },
-            "text": "text"
-          }
+          "value": [
+            {
+              "cell_id": "cell_id",
+              "location": {
+                "begin": 0,
+                "end": 1
+              },
+              "text": "text"
+            }
+          ]
         }
       ],
       "title": {

compare-comply/src/test/resources/compare_comply/table-return.json

@@ -127,14 +127,16 @@
             },
             "text": "text"
           },
-          "value": {
-            "cell_id": "cell_id",
-            "location": {
-              "begin": 0,
-              "end": 1
-            },
-            "text": "text"
-          }
+          "value": [
+            {
+              "cell_id": "cell_id",
+              "location": {
+                "begin": 0,
+                "end": 1
+              },
+              "text": "text"
+            }
+          ]
         }
       ]
     }