test.it.unimi.dsi.big.mg4j.query.nodes.AbstractTermExpanderTest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mg4j-big Show documentation

MG4J (Managing Gigabytes for Java) is a free full-text search engine for large document collections written in Java. The big version is a fork of the original MG4J that can handle more than 2^31 terms and documents.

The newest version!

package it.unimi.dsi.big.mg4j.query.nodes;

import static org.junit.Assert.assertEquals;
import it.unimi.dsi.big.mg4j.query.parser.QueryParserException;
import it.unimi.dsi.big.mg4j.query.parser.SimpleParser;

import org.junit.Test;


public class AbstractTermExpanderTest {

	@Test
	public void testExpand() throws QueryParserException {
		QueryTransformer addAqueryTransformer = new AbstractTermExpander() {
			@Override
			public Query expand( Term term ) {
				return new Term( "a" + term.term );
			}

			@Override
			public Query expand( Prefix prefix ) {
				return prefix;
			}
		};
		
		Query query = new SimpleParser().parse( "foo AND bar" );
		assertEquals( new And( new Term( "afoo"), new Term( "abar") ), addAqueryTransformer.transform( query ) ); 

		QueryTransformer expandQueryTransformer = new AbstractTermExpander() {
			/** The visitor used by this expander. */
			@Override
			public Query expand( Term term ) {
				return new MultiTerm( term, new Term( "a" + term.term ) );
			}

			@Override
			public Query expand( Prefix prefix ) {
				return prefix;
			}
		};
		
		assertEquals( new And( new MultiTerm( new Term( "foo" ), new Term( "afoo" ) ), new MultiTerm( new Term( "bar" ), new Term( "abar") ) ), expandQueryTransformer.transform( query ) ); 
	}
}