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