diff --git a/core/src/main/scala/org/apache/spark/api/java/JavaDoubleRDD.scala b/core/src/main/scala/org/apache/spark/api/java/JavaDoubleRDD.scala index 70f7f01d2beda8058e9ac346dad91d86706232b4..dad5c72e1c51a7e35f4ebc55757fb8c4e8507500 100644 --- a/core/src/main/scala/org/apache/spark/api/java/JavaDoubleRDD.scala +++ b/core/src/main/scala/org/apache/spark/api/java/JavaDoubleRDD.scala @@ -191,8 +191,8 @@ class JavaDoubleRDD(val srdd: RDD[scala.Double]) extends JavaRDDLike[Double, Jav * the maximum value of the last position and all NaN entries will be counted * in that bucket. */ - def histogram(buckets: Array[Double]): Array[Long] = { - srdd.histogram(buckets.map(_.toDouble), false) + def histogram(buckets: Array[scala.Double]): Array[Long] = { + srdd.histogram(buckets, false) } def histogram(buckets: Array[Double], evenBuckets: Boolean): Array[Long] = { diff --git a/core/src/test/scala/org/apache/spark/JavaAPISuite.java b/core/src/test/scala/org/apache/spark/JavaAPISuite.java index 8a9c6e63e0dd094043d98d3e9a7e418f88669b89..44483fd4ab99ba2ef3861ab359cf4a3a4615cdc5 100644 --- a/core/src/test/scala/org/apache/spark/JavaAPISuite.java +++ b/core/src/test/scala/org/apache/spark/JavaAPISuite.java @@ -368,10 +368,10 @@ public class JavaAPISuite implements Serializable { public void javaDoubleRDDHistoGram() { JavaDoubleRDD rdd = sc.parallelizeDoubles(Arrays.asList(1.0, 2.0, 3.0, 4.0)); // Test using generated buckets - Tuple2<Double[], long[]> results = rdd.histogram(2); - Double[] expected_buckets = {1.0, 2.5, 4.0}; + Tuple2<double[], long[]> results = rdd.histogram(2); + double[] expected_buckets = {1.0, 2.5, 4.0}; long[] expected_counts = {2, 2}; - Assert.assertArrayEquals(expected_buckets, results._1); + Assert.assertArrayEquals(expected_buckets, results._1, 0.1); Assert.assertArrayEquals(expected_counts, results._2); // Test with provided buckets long[] histogram = rdd.histogram(expected_buckets);