diff options
author | Jon Bratseth <bratseth@gmail.com> | 2021-12-17 12:49:32 +0100 |
---|---|---|
committer | Jon Bratseth <bratseth@gmail.com> | 2021-12-17 12:49:32 +0100 |
commit | 468ebc9a5527eaff02207443f7240e6da21fa7ac (patch) | |
tree | 2bd1d2844512765918c56caf86e7d534e62b3024 /linguistics-components/src/test/java/com/yahoo/language/tools/EmbedderTester.java | |
parent | 601b117281b74a578126a0f3effead55bc79c680 (diff) |
Test segmentation with subwords
Diffstat (limited to 'linguistics-components/src/test/java/com/yahoo/language/tools/EmbedderTester.java')
-rw-r--r-- | linguistics-components/src/test/java/com/yahoo/language/tools/EmbedderTester.java | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/linguistics-components/src/test/java/com/yahoo/language/tools/EmbedderTester.java b/linguistics-components/src/test/java/com/yahoo/language/tools/EmbedderTester.java index 9599e60e556..638efba2480 100644 --- a/linguistics-components/src/test/java/com/yahoo/language/tools/EmbedderTester.java +++ b/linguistics-components/src/test/java/com/yahoo/language/tools/EmbedderTester.java @@ -9,6 +9,7 @@ import com.yahoo.tensor.Tensor; import com.yahoo.tensor.TensorType; import java.util.Arrays; +import java.util.List; import static org.junit.Assert.assertArrayEquals; import static org.junit.Assert.assertEquals; @@ -53,7 +54,9 @@ public class EmbedderTester { } public void assertSegmented(Language language, String input, String... expectedSegments) { - assertArrayEquals(expectedSegments, ((Segmenter)embedder).segment(input, language).toArray()); + List<String> segments = ((Segmenter)embedder).segment(input, language); + assertArrayEquals("Actual segments: " + segments, + expectedSegments, ((Segmenter)embedder).segment(input, language).toArray()); } } |