def initialize_probability(self): '''Initialize self.pv to delta-distribution at v=0.''' # Delta initial probability distribution: self.pv = np.zeros_like(self.edges[:-1]) zero_bin_list = util.get_zero_bin_list(self.edges) for ii in zero_bin_list: self.pv[ii] = 1. / len(zero_bin_list)
def test_get_zero_bin_list(): assert util.get_zero_bin_list([0, 1, 2]) == [0] assert util.get_zero_bin_list([-1, 0, 1]) == [0, 1] assert util.get_zero_bin_list([-2, -1, 1]) == [1]
def test_get_zero_bin_list(): assert util.get_zero_bin_list([0,1,2]) == [0] assert util.get_zero_bin_list([-1,0,1]) == [0,1] assert util.get_zero_bin_list([-2,-1,1]) == [1]