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_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)
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)
