@Test
public void getProbability(){

	Node node = new LeafNode("ham", null);
	BiMap<String, Node> entityRegistry = ImmutableBiMap.of("1", node);
	NodeScoreDistribution<Double> classification = new NodeScoreDistribution<Double>(new ValueMap<Object, Double>(), node){
		@Override
		public BiMap<String, Node> getEntityRegistry(){
			return entityRegistry;
		}
		@Override
		public List<Node> getDecisionPath(){
			throw new UnsupportedOperationException();
		}
	};
	classification.put("ham", new DoubleValue(0.75d));
	classification.put("spam", new DoubleValue(0.25d));
	assertEquals(ImmutableSet.of("ham", "spam"), classification.getCategories());
	assertEquals((Double)0.75d, classification.getProbability("ham"));
	assertEquals((Double)0.25d, classification.getProbability("spam"));
}