Add tests for the Java implementation.
This commit is contained in:
parent
2a37235825
commit
e58c69d955
|
@ -364,6 +364,20 @@ public class JavaAPISuite implements Serializable {
|
|||
List<Double> take = rdd.take(5);
|
||||
}
|
||||
|
||||
@Test
|
||||
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};
|
||||
long[] expected_counts = {2, 2};
|
||||
Assert.assertArrayEquals(expected_buckets, results._1);
|
||||
Assert.assertArrayEquals(expected_counts, results._2);
|
||||
// Test with provided buckets
|
||||
long[] histogram = rdd.histogram(expected_buckets);
|
||||
Assert.assertArrayEquals(expected_counts, histogram);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void map() {
|
||||
JavaRDD<Integer> rdd = sc.parallelize(Arrays.asList(1, 2, 3, 4, 5));
|
||||
|
|
Loading…
Reference in a new issue