CPTTest.java

package aima.test.core.unit.probability.bayes.impl;

import org.junit.Assert;
import org.junit.Test;

import aima.core.probability.ProbabilityModel;
import aima.core.probability.bayes.impl.CPT;
import aima.core.probability.domain.BooleanDomain;
import aima.core.probability.util.RandVar;

public class CPTTest {
	public static final double DELTA_THRESHOLD = ProbabilityModel.DEFAULT_ROUNDING_THRESHOLD;

	@Test
	public void test_getConditioningCase() {
		RandVar aRV = new RandVar("A", new BooleanDomain());
		RandVar bRV = new RandVar("B", new BooleanDomain());
		RandVar cRV = new RandVar("C", new BooleanDomain());

		CPT cpt = new CPT(cRV, new double[] {
				// A = true, B = true, C = true
				0.1,
				// A = true, B = true, C = false
				0.9,
				// A = true, B = false, C = true
				0.2,
				// A = true, B = false, C = false
				0.8,
				// A = false, B = true, C = true
				0.3,
				// A = false, B = true, C = false
				0.7,
				// A = false, B = false, C = true
				0.4,
				// A = false, B = false, C = false
				0.6 }, aRV, bRV);

		Assert.assertArrayEquals(new double[] { 0.1, 0.9 }, cpt
				.getConditioningCase(true, true).getValues(), DELTA_THRESHOLD);

		Assert.assertArrayEquals(new double[] { 0.2, 0.8 }, cpt
				.getConditioningCase(true, false).getValues(), DELTA_THRESHOLD);

		Assert.assertArrayEquals(new double[] { 0.3, 0.7 }, cpt
				.getConditioningCase(false, true).getValues(), DELTA_THRESHOLD);

		Assert.assertArrayEquals(new double[] { 0.4, 0.6 }, cpt
				.getConditioningCase(false, false).getValues(), DELTA_THRESHOLD);

	}
}