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