[HIVEMALL-162] Test cases of L1/L2 normalize UDF expect HiveException

apache · Dec 19, 2017 · e2666ec · e2666ec
1 parent feb349a
commit e2666ec
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/core/src/test/java/hivemall/ftvec/scaling/L1NormalizationUDFTest.java b/core/src/test/java/hivemall/ftvec/scaling/L1NormalizationUDFTest.java
@@ -20,6 +20,7 @@
 
 import hivemall.utils.hadoop.WritableUtils;
 import hivemall.utils.math.MathUtils;
+import org.apache.hadoop.hive.ql.metadata.HiveException;
 import org.apache.hadoop.io.Text;
 import org.junit.Test;
 
@@ -31,7 +32,7 @@
 public class L1NormalizationUDFTest {
 
     @Test
-    public void test() {
+    public void test() throws HiveException {
         L1NormalizationUDF udf = new L1NormalizationUDF();
 
         assertEquals(null, udf.evaluate(null));

diff --git a/core/src/test/java/hivemall/ftvec/scaling/L2NormalizationUDFTest.java b/core/src/test/java/hivemall/ftvec/scaling/L2NormalizationUDFTest.java
@@ -25,13 +25,14 @@
 import java.util.Collections;
 import java.util.List;
 
+import org.apache.hadoop.hive.ql.metadata.HiveException;
 import org.apache.hadoop.io.Text;
 import org.junit.Test;
 
 public class L2NormalizationUDFTest {
 
     @Test
-    public void test() {
+    public void test() throws HiveException {
         L2NormalizationUDF udf = new L2NormalizationUDF();
 
         assertEquals(null, udf.evaluate(null));