natas_hw
/
jidt
mirror of https://github.com/jlizier/jidt
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Adding 3x XOR test case for MI discrete

This commit is contained in:
joseph.lizier 2012-09-05 06:37:13 +00:00
parent 65c7ac49a6
commit aa991901f6
1 changed files with 56 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
java/unittests/infodynamics/measures/discrete/MutualInformationTester.java
View File

@ -12,6 +12,14 @@ public class MutualInformationTester extends TestCase {
miCalc.addObservations(new int[] {0, 0, 1, 1}, new int[] {0, 0, 1, 1});
double miCopy = miCalc.computeAverageLocalOfObservations();
assertEquals(1.0, miCopy, 0.000001);
// check that the number of observations doesn't cause any changes here:
for (int n = 1; n < 100; n++) {
miCalc.addObservations(new int[] {0, 0, 1, 1}, new int[] {0, 0, 1, 1});
assertEquals(4*(n+1), miCalc.getNumObservations());
miCopy = miCalc.computeAverageLocalOfObservations();
assertEquals(1.0, miCopy, 0.000001);
}
}
public void testIndependent() {
@ -43,13 +51,59 @@ public class MutualInformationTester extends TestCase {
double miX2Y = miCalc.computeAverageLocalOfObservations();
assertEquals(0.0, miX2Y, 0.000001);
// Y is fully determined from X1, X2 - MI should be 0 bits
// Y is fully determined from X1, X2 - MI should be 1 bits
MutualInformationCalculator miCalcBase4 = new MutualInformationCalculator(4, 0);
int[] X12 = new int[] {0, 1, 2, 3};
miCalcBase4.initialise();
miCalcBase4.addObservations(X12, Y);
miCalcBase4.setDebug(true);
// miCalcBase4.setDebug(true);
double miX12Y = miCalcBase4.computeAverageLocalOfObservations();
assertEquals(1.0, miX12Y, 0.000001);
}
public void test3Xor() {
MutualInformationCalculator miCalc = new MutualInformationCalculator(2, 0);
int[] X1 = new int[] {0, 1, 0, 1, 0, 1, 0, 1};
int[] X2 = new int[] {0, 0, 1, 1, 0, 0, 1, 1};
int[] X3 = new int[] {0, 0, 0, 0, 1, 1, 1, 1};
int[] Y = new int[] {0, 1, 1, 0, 1, 0, 0, 1};
// Y is independent of X1 - MI should be 0 bits
miCalc.initialise();
miCalc.addObservations(X1, Y);
double miX1Y = miCalc.computeAverageLocalOfObservations();
assertEquals(0.0, miX1Y, 0.000001);
// Y is independent of X2 - MI should be 0 bits
miCalc.initialise();
miCalc.addObservations(X2, Y);
double miX2Y = miCalc.computeAverageLocalOfObservations();
assertEquals(0.0, miX2Y, 0.000001);
// Y is independent of X3 - MI should be 0 bits
miCalc.initialise();
miCalc.addObservations(X3, Y);
double miX3Y = miCalc.computeAverageLocalOfObservations();
assertEquals(0.0, miX3Y, 0.000001);
// Y is independent of X1, X2 - MI should be 0 bits
MutualInformationCalculator miCalcBase4 = new MutualInformationCalculator(4, 0);
int[] X12 = new int[] {0, 1, 2, 3, 0, 1, 2, 3};
miCalcBase4.initialise();
miCalcBase4.addObservations(X12, Y);
// miCalcBase4.setDebug(true);
double miX12Y = miCalcBase4.computeAverageLocalOfObservations();
assertEquals(0.0, miX12Y, 0.000001);
// Y is fully determined from X1, X2, X3 - MI should be 1 bits
MutualInformationCalculator miCalcBase8 = new MutualInformationCalculator(8, 0);
int[] X123 = new int[] {0, 1, 2, 3, 4, 5, 6, 7};
miCalcBase8.initialise();
miCalcBase8.addObservations(X123, Y);
// miCalcBase8.setDebug(true);
double miX123Y = miCalcBase8.computeAverageLocalOfObservations();
assertEquals(1.0, miX123Y, 0.000001);
}
}