Examples with IndexArgs io.anserini.index.IndexArgs used on opensource projects

Example 16 with IndexArgs

use of io.anserini.index.IndexArgs in project Anserini by castorini.

the class AclAnthologyEndToEndTest method getIndexArgs.

@Override
public IndexArgs getIndexArgs() {
    IndexArgs indexArgs = createDefaultIndexArgs();
    indexArgs.input = "src/test/resources/sample_docs/acl";
    indexArgs.collectionClass = AclAnthology.class.getSimpleName();
    indexArgs.generatorClass = AclAnthologyGenerator.class.getSimpleName();
    return indexArgs;
}

Example 17 with IndexArgs

use of io.anserini.index.IndexArgs in project Anserini by castorini.

the class SolrEndToEndTest method testIndexAndSearch.

@Test
public void testIndexAndSearch() throws Exception {
    String schemaAdjustmentFile = getSchemaAdjustmentFile();
    if (schemaAdjustmentFile != null) {
        // update schema, much like curl -X POST -H 'Content-type:application/json' --data-binary SCHEMA_NAME.json http://localhost:8983/solr/COLLECTION_NAME/schema
        String schemaJson = Files.readString(getFile(schemaAdjustmentFile).toPath());
        DirectJsonQueryRequest schemaRequest = new DirectJsonQueryRequest(schemaJson, params(CommonParams.QT, "/schema"));
        QueryResponse response = schemaRequest.process(client, getCollectionName());
        assertEquals(0, response.getStatus());
    }
    IndexArgs indexArgs = getIndexArgs();
    IndexCollection indexRunner = getIndexRunner(indexArgs);
    indexRunner.run();
    SearchSolr.Args searchArgs = getSearchArgs();
    SearchSolr searchRunner = getSearchRunner(searchArgs);
    searchRunner.runTopics();
    BufferedReader br = new BufferedReader(new FileReader(searchArgs.output));
    String[] ref = getRefRankingResult();
    String s;
    int cnt = 0;
    while ((s = br.readLine()) != null) {
        assertEquals(ref[cnt], s);
        cnt++;
    }
    assertEquals(cnt, ref.length);
    FileUtils.deleteQuietly(new File(searchArgs.output));
}

Example 18 with IndexArgs

use of io.anserini.index.IndexArgs in project Anserini by castorini.

the class SolrEndToEndTest method createDefaultIndexArgs.

protected IndexArgs createDefaultIndexArgs() {
    IndexArgs args = new IndexArgs();
    args.solrIndex = getCollectionName();
    args.threads = 1;
    args.storePositions = true;
    args.storeDocvectors = true;
    args.storeContents = true;
    args.storeRaw = true;
    args.optimize = true;
    args.quiet = true;
    args.solr = true;
    return args;
}

Example 19 with IndexArgs

use of io.anserini.index.IndexArgs in project Anserini by castorini.

the class AclAnthologyEndToEndTest method getIndexArgs.

@Override
protected IndexArgs getIndexArgs() {
    IndexArgs indexArgs = createDefaultIndexArgs();
    indexArgs.input = "src/test/resources/sample_docs/acl";
    indexArgs.collectionClass = AclAnthology.class.getSimpleName();
    indexArgs.generatorClass = AclAnthologyGenerator.class.getSimpleName();
    return indexArgs;
}

Example 20 with IndexArgs

use of io.anserini.index.IndexArgs in project Anserini by castorini.

the class MultiThreadingSearchTest method getIndexArgs.

@Override
protected IndexArgs getIndexArgs() {
    IndexArgs indexArgs = createDefaultIndexArgs();
    indexArgs.input = "src/test/resources/sample_docs/trec/collection2";
    indexArgs.collectionClass = TrecCollection.class.getSimpleName();
    return indexArgs;
}