com.acmutv.ontoqa.benchmark.basic.QuestionB01Test.java Source code

Java tutorial

Introduction

Here is the source code for com.acmutv.ontoqa.benchmark.basic.QuestionB01Test.java

Source

/*
  The MIT License (MIT)
    
  Copyright (c) 2016 Antonella Botte, Giacomo Marciani and Debora Partigianoni
    
  Permission is hereby granted, free of charge, to any person obtaining a copy
  of this software and associated documentation files (the "Software"), to deal
  in the Software without restriction, including without limitation the rights
  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  copies of the Software, and to permit persons to whom the Software is
  furnished to do so, subject to the following conditions:
    
    
  The above copyright notice and this permission notice shall be included in
  all copies or substantial portions of the Software.
    
    
  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  THE SOFTWARE.
 */

package com.acmutv.ontoqa.benchmark.basic;

import com.acmutv.ontoqa.benchmark.Common;
import com.acmutv.ontoqa.core.CoreController;
import com.acmutv.ontoqa.core.exception.*;
import com.acmutv.ontoqa.core.grammar.CommonGrammar;
import com.acmutv.ontoqa.core.grammar.Grammar;
import com.acmutv.ontoqa.core.knowledge.answer.Answer;
import com.acmutv.ontoqa.core.knowledge.answer.SimpleAnswer;
import com.acmutv.ontoqa.core.knowledge.ontology.Ontology;
import com.acmutv.ontoqa.core.semantics.dudes.DudesTemplates;
import com.acmutv.ontoqa.core.semantics.sltag.SimpleSltag;
import com.acmutv.ontoqa.core.semantics.sltag.Sltag;
import com.acmutv.ontoqa.core.semantics.sltag.SltagBuilder;
import com.acmutv.ontoqa.core.syntax.ltag.LtagTemplates;
import org.apache.commons.lang3.tuple.Pair;
import org.apache.jena.query.Query;
import org.apache.jena.query.QueryFactory;
import org.apache.jena.query.Syntax;
import org.junit.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;

import static com.acmutv.ontoqa.benchmark.Common.*;

/**
 * JUnit tests for questions of class [CLASS BASIC-1].
 * `Who founded Microsoft?`
 * @author Antonella Botte {@literal <abotte@acm.org>}
 * @author Giacomo Marciani {@literal <gmarciani@acm.org>}
 * @author Debora Partigianoni {@literal <dpartigianoni@acm.org>}
 * @since 1.0
 */
public class QuestionB01Test {

    private static final Logger LOGGER = LoggerFactory.getLogger(QuestionB01Test.class);

    private static final String QUESTION = "Who founded Microsoft?";

    private static final Answer ANSWER = new SimpleAnswer(BILL_GATES_IRI, PAUL_ALLEN_IRI);

    private static final String QUERY_1 = String.format(
            "SELECT DISTINCT  ?v4\n" + "WHERE\n" + "  { <%s>\n" + "              <%s>  ?v4}\n", MICROSOFT_IRI,
            HAS_FOUNDER_IRI);

    private static final String QUERY_2 = String.format(
            "SELECT DISTINCT  ?v5\n" + "WHERE\n" + "  { <%s>\n" + "              <%s>  ?v5}\n", MICROSOFT_IRI,
            HAS_FOUNDER_IRI);

    /**
     * Tests the question-answering with parsing.
     * @throws QuestionException when the question is malformed.
     * @throws OntoqaFatalException when the question cannot be processed due to some fatal errors.
     */
    @Test
    public void test_nlp() throws Exception {
        final Grammar grammar = Common.getGrammar();
        final Ontology ontology = Common.getOntology();
        final Pair<Query, Answer> result = CoreController.process(QUESTION, grammar, ontology);
        final Query query = result.getKey();
        final Answer answer = result.getValue();
        LOGGER.info("Query: {}", query);
        LOGGER.info("Answer: {}", answer);
        Assert.assertEquals(QUERY_1, query.toString());
        Assert.assertEquals(ANSWER, answer);
    }

    /**
     * Tests the question-answering with parsing.
     * @throws QuestionException when the question is malformed.
     * @throws OntoqaFatalException when the question cannot be processed due to some fatal errors.
     */
    @Test
    public void test_nlp_wired() throws Exception {
        final Grammar grammar = CommonGrammar.build_completeGrammar();
        final Ontology ontology = Common.getOntology();
        final Pair<Query, Answer> result = CoreController.process(QUESTION, grammar, ontology);
        final Query query = result.getKey();
        final Answer answer = result.getValue();
        LOGGER.info("Query: {}", query);
        LOGGER.info("Answer: {}", answer);
        Assert.assertEquals(QUERY_2, query.toString());
        Assert.assertEquals(ANSWER, answer);
    }

    /**
     * Tests the question-answering with manual compilation of SLTAG.
     * @throws QuestionException when question is malformed.
     * @throws OntoqaFatalException when question cannot be processed due to some fatal errors.
     */
    @Test
    public void test_manual()
            throws OntoqaFatalException, QuestionException, QueryException, LTAGException, IOException {
        /* who */
        Sltag who = new SimpleSltag(LtagTemplates.wh("who"), DudesTemplates.who());
        LOGGER.info("who:\n{}", who.toPrettyString());

        /* founded */
        Sltag founded = new SimpleSltag(
                LtagTemplates.transitiveVerbActiveIndicative("founded", "person", "company"),
                DudesTemplates.property(HAS_FOUNDER_IRI, "company", "person"));
        LOGGER.info("founded:\n{}", founded.toPrettyString());

        /* Microsoft */
        Sltag microsoft = new SimpleSltag(LtagTemplates.properNoun("Microsoft"),
                DudesTemplates.properNoun(MICROSOFT_IRI));
        LOGGER.info("Microsoft:\n{}", microsoft.toPrettyString());

        /* who founded Microsoft */
        LOGGER.info("who founded Microsoft: processing...");
        Sltag whoFoundedMicrosoft = new SltagBuilder(founded).substitution(who, "person")
                .substitution(microsoft, "company").build();
        LOGGER.info("who founded Microsoft:\n{}", whoFoundedMicrosoft.toPrettyString());

        /* SPARQL */
        LOGGER.info("SPARQL query: processing...");
        Query query = whoFoundedMicrosoft.convertToSPARQL();
        LOGGER.info("SPARQL query:\n{}", query);

        Common.test_query(query, ANSWER);
    }

    /**
     * Tests the ontology answering on raw SPARQL query submission.
     */
    @Test
    public void test_ontology() throws OntoqaFatalException, IOException, QueryException {
        String sparql = String.format("SELECT ?x WHERE { <%s> <%s> ?x }", MICROSOFT_IRI, HAS_FOUNDER_IRI);
        Query query = QueryFactory.create(sparql);
        LOGGER.debug("SPARQL query:\n{}", query);
        Common.test_query(query, ANSWER);
    }
}