コード例 #1
0
  @Test
  public void testGetAlignmentSets() throws Exception {
    AMR amr = new AMR();
    AMR.Node root = amr.addNode("r", "root");
    AMR.Node child1 = amr.addNode("c1", "child1");
    AMR.Node child2 = amr.addNode("c2", "child2");
    AMR.Arc arc1 = amr.addArc(root, child1, "arc1");
    AMR.Arc arc2 = amr.addArc(root, child2, "arc2");

    child1.alignment = 0;
    root.alignment = 1;
    child2.alignment = 2;

    Map<Integer, Set<AMR.Node>> alignmentSets = ParserUtils.getAlignmentSets(amr);
    assertEquals(1, alignmentSets.get(0).size());
    assertEquals(child1, alignmentSets.get(0).iterator().next());
    assertEquals(1, alignmentSets.get(1).size());
    assertEquals(root, alignmentSets.get(1).iterator().next());
    assertEquals(1, alignmentSets.get(2).size());
    assertEquals(child2, alignmentSets.get(2).iterator().next());

    child2.alignment = 0;

    alignmentSets = ParserUtils.getAlignmentSets(amr);
    assertEquals(2, alignmentSets.get(0).size());
    assertTrue(alignmentSets.get(0).contains(child1));
    assertTrue(alignmentSets.get(0).contains(child2));
    assertEquals(1, alignmentSets.get(1).size());
    assertEquals(root, alignmentSets.get(1).iterator().next());
  }
コード例 #2
0
  @Test
  public void testAlignmentChunksAndArcs() throws Exception {
    AMR amr = new AMR();
    AMR.Node root = amr.addNode("r", "root");
    AMR.Node child1 = amr.addNode("c1", "child1");
    AMR.Node child2 = amr.addNode("c2", "child2");
    AMR.Arc arc1 = amr.addArc(root, child1, "arc1");
    AMR.Arc arc2 = amr.addArc(root, child2, "arc2");
    amr.sourceText = new String[] {"he", "ran", "dog"};

    child1.alignment = 0;
    root.alignment = 1;
    child2.alignment = 2;

    Pair<AMR[], List<ParserUtils.ChunkArc>> pair = ParserUtils.alignmentChunksAndArcs(amr);
    AMR[] chunks = pair.first;
    List<ParserUtils.ChunkArc> chunkArcs = pair.second;

    assertEquals(3, chunks.length);
    assertEquals(2, chunkArcs.size());
  }
コード例 #3
0
  @Test
  public void testIsAlignmentChunkable() throws Exception {
    AMR amr = new AMR();
    AMR.Node root = amr.addNode("r", "root");
    AMR.Node child1 = amr.addNode("c1", "child1");
    AMR.Node child2 = amr.addNode("c2", "child2");
    AMR.Arc arc1 = amr.addArc(root, child1, "arc1");
    AMR.Arc arc2 = amr.addArc(root, child2, "arc2");

    child1.alignment = 0;
    root.alignment = 1;
    child2.alignment = 2;

    assertEquals(true, ParserUtils.isAlignmentChunkable(amr));

    child2.alignment = 0;
    assertEquals(false, ParserUtils.isAlignmentChunkable(amr));

    child1.alignment = 1;
    assertEquals(true, ParserUtils.isAlignmentChunkable(amr));
  }