def test_accumulate_no_weights(self):
"""
"""
expected_h_tpl = np.array([2, 3, 2, 2, 2])
expected_c_tpl = np.array([-700.7, -0.5, 0.01, 300.3, 500.5])
expected_h = np.zeros(shape=self.n_bins, dtype=np.double)
expected_c = np.zeros(shape=self.n_bins, dtype=np.double)
self.fill_histo(expected_h, expected_h_tpl, self.ndims-1)
self.fill_histo(expected_c, expected_c_tpl, self.ndims-1)
histo_inst = Histogramnd(self.sample,
self.histo_range,
self.n_bins,
weights=self.weights)
sample_2 = self.sample[:]
if len(sample_2.shape) == 1:
idx = [slice(0, None)]
else:
idx = [slice(0, None), self.tested_dim]
sample_2[idx] += 2
histo_inst.accumulate(sample_2) # <==== !!
histo = histo_inst.histo
cumul = histo_inst.weighted_histo
bin_edges = histo_inst.edges
self.assertEqual(cumul.dtype, np.float64)
self.assertTrue(np.array_equal(histo, expected_h))
self.assertTrue(np.allclose(cumul, expected_c, rtol=10e-15))
def test_accumulate_no_weights(self):
"""
"""
expected_h_tpl = np.array([2, 3, 2, 2, 2])
expected_c_tpl = np.array([-700.7, -0.5, 0.01, 300.3, 500.5])
expected_h = np.zeros(shape=self.n_bins, dtype=np.double)
expected_c = np.zeros(shape=self.n_bins, dtype=np.double)
self.fill_histo(expected_h, expected_h_tpl, self.ndims - 1)
self.fill_histo(expected_c, expected_c_tpl, self.ndims - 1)
histo_inst = Histogramnd(self.sample,
self.histo_range,
self.n_bins,
weights=self.weights)
sample_2 = self.sample[:]
if len(sample_2.shape) == 1:
idx = (slice(0, None), )
else:
idx = slice(0, None), self.tested_dim
sample_2[idx] += 2
histo_inst.accumulate(sample_2) # <==== !!
histo = histo_inst.histo
cumul = histo_inst.weighted_histo
bin_edges = histo_inst.edges
self.assertEqual(cumul.dtype, np.float64)
self.assertTrue(np.array_equal(histo, expected_h))
self.assertTrue(np.allclose(cumul, expected_c, rtol=10e-15))
def test_empty_init_accumulate(self):
"""
"""
expected_h_tpl = np.array([2, 1, 1, 1, 1])
expected_c_tpl = np.array([-700.7, -0.5, 0.01, 300.3, 500.5])
expected_h = np.zeros(shape=self.n_bins, dtype=np.double)
expected_c = np.zeros(shape=self.n_bins, dtype=np.double)
self.fill_histo(expected_h, expected_h_tpl, self.ndims - 1)
self.fill_histo(expected_c, expected_c_tpl, self.ndims - 1)
histo_inst = Histogramnd(None, self.histo_range, self.n_bins)
histo_inst.accumulate(self.sample, weights=self.weights)
histo = histo_inst.histo
cumul = histo_inst.weighted_histo
bin_edges = histo_inst.edges
expected_edges = _get_bin_edges(self.histo_range, self.n_bins,
self.ndims)
self.assertEqual(cumul.dtype, np.float64)
self.assertEqual(histo.dtype, np.uint32)
self.assertTrue(np.array_equal(histo, expected_h))
self.assertTrue(np.array_equal(cumul, expected_c))
for i_edges, edges in enumerate(expected_edges):
self.assertTrue(np.array_equal(bin_edges[i_edges],
expected_edges[i_edges]),
msg='Testing bin_edges for dim {0}'
''.format(i_edges + 1))
def test_empty_init_accumulate(self):
"""
"""
expected_h_tpl = np.array([2, 1, 1, 1, 1])
expected_c_tpl = np.array([-700.7, -0.5, 0.01, 300.3, 500.5])
expected_h = np.zeros(shape=self.n_bins, dtype=np.double)
expected_c = np.zeros(shape=self.n_bins, dtype=np.double)
self.fill_histo(expected_h, expected_h_tpl, self.ndims-1)
self.fill_histo(expected_c, expected_c_tpl, self.ndims-1)
histo_inst = Histogramnd(None,
self.histo_range,
self.n_bins)
histo_inst.accumulate(self.sample,
weights=self.weights)
histo = histo_inst.histo
cumul = histo_inst.weighted_histo
bin_edges = histo_inst.edges
expected_edges = _get_bin_edges(self.histo_range,
self.n_bins,
self.ndims)
self.assertEqual(cumul.dtype, np.float64)
self.assertEqual(histo.dtype, np.uint32)
self.assertTrue(np.array_equal(histo, expected_h))
self.assertTrue(np.array_equal(cumul, expected_c))
for i_edges, edges in enumerate(expected_edges):
self.assertTrue(np.array_equal(bin_edges[i_edges],
expected_edges[i_edges]),
msg='Testing bin_edges for dim {0}'
''.format(i_edges+1))
