nlp-utils/src/test/java/org/apache/opennlp/utils/TestUtils.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.opennlp.utils;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Random;

/**
 * Utility class for tests
 */
public class TestUtils {

  private static final Random R = new Random();

  public static void fillTrainingSet(TrainingSet trainingSet, int size, int dimension) {
    for (int i = 0; i < size; i++) {
      double[] inputs = new double[dimension];
      for (int j = 0; j < dimension; j++) {
        inputs[j] = Math.random();
      }
      double out = Math.random();
      trainingSet.add(new TrainingExample(inputs, out));
    }
  }

  public static Collection<String[]> generateRandomVocabulary() {
    int size = R.nextInt(1000);
    Collection<String[]> vocabulary = new ArrayList<>(size);
    for (int i = 0; i < size; i++) {
      String[] sentence = generateRandomSentence();
      vocabulary.add(sentence);
    }
    return vocabulary;
  }

  public static String[] generateRandomSentence() {
    int dimension = R.nextInt(10);
    String[] sentence = new String[dimension];
    for (int j = 0; j < dimension; j++) {
      char c = (char) R.nextInt(10);
      sentence[j] = c + "-" + c + "-" + c;
    }
    return sentence;
  }
}