com.bigdata.rdf.graph.impl.bd.TestSSSP Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bigdata-rdf-test Show documentation
Blazegraph(TM) RDF Test Suites
There is a newer version: 2.1.4
/**

Copyright (C) SYSTAP, LLC DBA Blazegraph 2006-2016.  All rights reserved.

Contact:
     SYSTAP, LLC DBA Blazegraph
     2501 Calvert ST NW #106
     Washington, DC 20008
     [email protected]

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; version 2 of the License.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/
package com.bigdata.rdf.graph.impl.bd;

import org.openrdf.model.URI;

import com.bigdata.rdf.graph.IGASContext;
import com.bigdata.rdf.graph.IGASEngine;
import com.bigdata.rdf.graph.IGASState;
import com.bigdata.rdf.graph.IGraphAccessor;
import com.bigdata.rdf.graph.analytics.SSSP;

/**
 * Test class for SSP traversal.
 * 
 * @author Bryan Thompson
 */
public class TestSSSP extends AbstractBigdataGraphTestCase {

    public TestSSSP() {
        
    }
    
    public TestSSSP(String name) {
        super(name);
    }

    public void testSSSP() throws Exception {

        final SmallGraphProblem p = setupSmallGraphProblem();

        final IGASEngine gasEngine = getGraphFixture()
                .newGASEngine(1/* nthreads */);

        try {

            final IGraphAccessor graphAccessor = getGraphFixture()
                    .newGraphAccessor(null/* ignored */);

            final IGASContext gasContext = gasEngine
                    .newGASContext(graphAccessor, new SSSP());

            final IGASState gasState = gasContext.getGASState();
            
            // Initialize the froniter.
            gasState.setFrontier(gasContext, p.getMike());

            // Converge.
            gasContext.call();

            assertEquals(0.0, gasState.getState(p.getMike()).dist());

            assertEquals(1.0, gasState.getState(p.getFoafPerson()).dist());

            assertEquals(1.0, gasState.getState(p.getBryan()).dist());

            assertEquals(2.0, gasState.getState(p.getMartyn()).dist());

        } finally {

            gasEngine.shutdownNow();

        }

    }
    
    /**
     * A unit test based on graph with link weights - in this version of the
     * test we constrain the link type but do not specify the link attribute
     * type. Hence it ignores the link weights. This provides a test of the
     * optimized access path when just the link type constraint is specified.
     */
    public void test_sssp_linkType_constraint() throws Exception {
        
        final SmallWeightedGraphProblem p = setupSmallWeightedGraphProblem();

        final IGASEngine gasEngine = getGraphFixture()
                .newGASEngine(1/* nthreads */);

        try {

            final IGraphAccessor graphAccessor = getGraphFixture()
                    .newGraphAccessor(null/* ignored */);

            final IGASContext gasContext = gasEngine
                    .newGASContext(graphAccessor, new SSSP());

            // Set constraint on the visited link types.
            gasContext.setLinkType((URI) p.getFoafKnows());

            final IGASState gasState = gasContext.getGASState();
            
            // Initialize the froniter.
            gasState.setFrontier(gasContext, p.getV1());

            // Converge.
            gasContext.call();

            // check #of hops.
            assertEquals(0.0, gasState.getState(p.getV1()).dist());

            assertEquals(1.0, gasState.getState(p.getV2()).dist());

            assertEquals(1.0, gasState.getState(p.getV3()).dist());

            assertEquals(2.0, gasState.getState(p.getV4()).dist());

            assertEquals(2.0, gasState.getState(p.getV5()).dist());

        } finally {

            gasEngine.shutdownNow();

        }

    }

    /**
     * A unit test based on graph with link weights.
     * 
     * FIXME Test with just the linkAttributeType constraint and with both a
     * linkType and linkAttributeType constraint. (We already have a test with
     * just the linkType constraint above).
     * 
     * FIXME This is only testing the scatter AP. We also need to test the
     * gather AP.
     */
    public void test_sssp_weightedGraph() throws Exception {
        
        final SmallWeightedGraphProblem p = setupSmallWeightedGraphProblem();

        final IGASEngine gasEngine = getGraphFixture()
                .newGASEngine(1/* nthreads */);

        try {

            final IGraphAccessor graphAccessor = getGraphFixture()
                    .newGraphAccessor(null/* ignored */);

            final IGASContext gasContext = gasEngine
                    .newGASContext(graphAccessor, new SSSP());

            // Set constraint on the visited link types.
            gasContext.setLinkType((URI) p.getFoafKnows());

            // Set constraint on the visited link attribute types.
            gasContext.setLinkAttributeType((URI) p.getLinkWeight());

            final IGASState gasState = gasContext.getGASState();
            
            // Initialize the froniter.
            gasState.setFrontier(gasContext, p.getV1());

            // Converge.
            gasContext.call();

            // Check weighted distance.
            assertEquals(0.0, gasState.getState(p.getV1()).dist());

            assertEquals(100.0, gasState.getState(p.getV2()).dist());

            assertEquals(100.0, gasState.getState(p.getV3()).dist());

            assertEquals(150.0, gasState.getState(p.getV4()).dist());

            assertEquals(175.0, gasState.getState(p.getV5()).dist());

        } finally {

            gasEngine.shutdownNow();

        }

    }

}