def test_ignore_out_of_range(self): data = self.sc.parallelize([1, 2, 3, 4, 5]) hist, bin_edges = histogram(data, range=(2, 5), bins=2) assert_equals(2, hist[0]) assert_equals(1, hist[1]) assert_array_almost_equal(np.array([2, 3.5, 5]), bin_edges)
def test_bins_as_array(self): data = self.sc.parallelize([1, 2, 3, 4, 5]) hist, bin_edges = histogram(data, bins=[0, 3, 6]) assert_equals(2, hist[0]) assert_equals(3, hist[1]) assert_array_almost_equal(np.array([0, 3, 6]), bin_edges)