From 08f25beee605cde4dfd46c857cb0a69dd570b458 Mon Sep 17 00:00:00 2001
From: Pedro Martinez Mediano <pedro.martinez-mediano13@imperial.ac.uk>
Date: Fri, 12 Jan 2018 22:23:12 +0000
Subject: [PATCH] Add tests for new overloadings in KSG mixed calc.

---
 ...MultiVariateWithDiscreteKraskovTester.java | 42 +++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/java/unittests/infodynamics/measures/mixed/kraskov/MutualInfoMultiVariateWithDiscreteKraskovTester.java b/java/unittests/infodynamics/measures/mixed/kraskov/MutualInfoMultiVariateWithDiscreteKraskovTester.java
index f692491..4bfdf96 100755
--- a/java/unittests/infodynamics/measures/mixed/kraskov/MutualInfoMultiVariateWithDiscreteKraskovTester.java
+++ b/java/unittests/infodynamics/measures/mixed/kraskov/MutualInfoMultiVariateWithDiscreteKraskovTester.java
@@ -269,4 +269,46 @@ public class MutualInfoMultiVariateWithDiscreteKraskovTester extends TestCase {
     // assertTrue(Math.abs(res5 - res1) > 0.001);
 
   }
+
+  public void testUnivariateOverloadings() throws Exception {
+    MutualInfoCalculatorMultiVariateWithDiscreteKraskov miCalc =
+        new MutualInfoCalculatorMultiVariateWithDiscreteKraskov();
+
+    // Generate data sets
+    RandomGenerator rg = new RandomGenerator();
+    double[][] contDataMatrix = rg.generateRandomData(100, 2);
+    double[] contDataVector   = rg.generateRandomData(100);
+    int[] discData = rg.generateRandomInts(100, 2);
+
+    // Check that no exception is thrown for ok data:
+    boolean caughtException = false;
+    try {
+      miCalc.initialise(1, 2);
+      miCalc.setObservations(contDataVector, discData);
+    } catch (Exception e) {
+      caughtException = true;
+    }
+    assertFalse(caughtException);
+
+    // Check that exception is thrown when dimensions do not match
+    caughtException = false;
+    try {
+      miCalc.initialise(1, 2);
+      miCalc.setObservations(contDataMatrix, discData);
+    } catch (Exception e) {
+      caughtException = true;
+    }
+    assertTrue(caughtException);
+
+    caughtException = false;
+    try {
+      miCalc.initialise(2, 2);
+      miCalc.setObservations(contDataVector, discData);
+    } catch (Exception e) {
+      caughtException = true;
+    }
+    assertTrue(caughtException);
+
+
+  }
 }