@Test public void testCreationWihtSimpleDescriptor() throws Exception { InputStream generatorDescriptorIn = getClass() .getResourceAsStream("/opennlp/tools/util/featuregen/TestFeatureGeneratorConfig.xml"); // If this fails the generator descriptor could not be found // at the expected location Assert.assertNotNull(generatorDescriptorIn); Collection<String> expectedGenerators = new ArrayList<String>(); expectedGenerators.add(OutcomePriorFeatureGenerator.class.getName()); AggregatedFeatureGenerator aggregatedGenerator = (AggregatedFeatureGenerator) GeneratorFactory.create(generatorDescriptorIn, null); for (AdaptiveFeatureGenerator generator : aggregatedGenerator.getGenerators()) { expectedGenerators.remove(generator.getClass().getName()); // if of kind which requires parameters check that } // If this fails not all expected generators were found and // removed from the expected generators collection Assert.assertEquals(0, expectedGenerators.size()); }
/** * Tests the creation from a descriptor which contains an unkown element. The creation should fail * with an {@link InvalidFormatException} */ @Test(expected = IOException.class) public void testCreationWithUnkownElement() throws IOException { InputStream descIn = getClass() .getResourceAsStream( "/opennlp/tools/util/featuregen/FeatureGeneratorConfigWithUnkownElement.xml"); try { GeneratorFactory.create(descIn, null); } finally { descIn.close(); } }
@Test public void testCreationWithCustomGenerator() throws Exception { InputStream generatorDescriptorIn = getClass().getResourceAsStream("/opennlp/tools/util/featuregen/CustomClassLoading.xml"); // If this fails the generator descriptor could not be found // at the expected location Assert.assertNotNull(generatorDescriptorIn); AggregatedFeatureGenerator aggregatedGenerator = (AggregatedFeatureGenerator) GeneratorFactory.create(generatorDescriptorIn, null); Collection<AdaptiveFeatureGenerator> embeddedGenerator = aggregatedGenerator.getGenerators(); Assert.assertEquals(1, embeddedGenerator.size()); for (AdaptiveFeatureGenerator generator : embeddedGenerator) { Assert.assertEquals(TokenFeatureGenerator.class.getName(), generator.getClass().getName()); } }