Tiny test of the UniversalPOSMapper

stanfordnlp · Mar 14, 2024 · fd6800f · fd6800f
1 parent 731fc8e
commit fd6800f
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/test/src/edu/stanford/nlp/trees/UniversalPOSMapperTest.java b/test/src/edu/stanford/nlp/trees/UniversalPOSMapperTest.java
@@ -0,0 +1,17 @@
+package edu.stanford.nlp.trees;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+import edu.stanford.nlp.trees.Tree;
+
+public class UniversalPOSMapperTest {
+  @Test
+  public void testMap() {
+    Tree tree = Tree.valueOf("(ROOT (S (NP (DT This)) (VP (VBZ is) (NP (DT a) (JJ simple) (NN test)))))");
+    Tree newTree = UniversalPOSMapper.mapTree(tree);
+    Tree expected = Tree.valueOf("(ROOT (S (NP (PRON This)) (VP (VERB is) (NP (DET a) (ADJ simple) (NOUN test)))))");
+    Assert.assertEquals(expected, newTree);
+  }
+}
+