@Override public Iterator<List<String>> getGeneratorImpl(GeneratorFactory generatorFactory) { Tuple2<Double, String> p1 = Tuple.tuple2(1.0, "1"); Tuple2<Double, String> p2 = Tuple.tuple2(1.0, "2"); Tuple2<Double, String> p3 = Tuple.tuple2(1.0, "3"); ArrayList<Tuple2<Double, String>> items = new ArrayList<Tuple2<Double, String>>(); items.add(p1); items.add(p2); items.add(p3); Iterator<Integer> amountToRetrieveGenerator = generatorFactory.uniform(0, 3); Iterator<List<String>> generator = generatorFactory.weightedDiscreteList(items, amountToRetrieveGenerator); return generator; }
@Test(expected = GeneratorException.class) public void emptyConstructorTest() { // Given GeneratorFactory generatorFactory = new GeneratorFactory(new RandomDataGeneratorFactory()); Iterator<Integer> amountToRetrieveGenerator = generatorFactory.uniform(0, 3); ArrayList<Tuple2<Double, String>> emptyItems = new ArrayList<Tuple2<Double, String>>(); Iterator<List<String>> generator = generatorFactory.weightedDiscreteList(emptyItems, amountToRetrieveGenerator); // When generator.next(); // Then assertEquals("Empty DiscreteGenerator should throw exception on next()", false, true); }