test_sampler.py 文件源码

def test_sample_prior(self):

        rnd1 = np.random.RandomState(42)
        joker1 = TheJoker(self.joker_params['binary'], random_state=rnd1)

        rnd2 = np.random.RandomState(42)
        joker2 = TheJoker(self.joker_params['triple'], random_state=rnd2)

        samples1 = joker1.sample_prior(8)
        samples2 = joker2.sample_prior(8)

        for key in samples1.keys():
            assert quantity_allclose(samples1[key], samples2[key])

        samples, ln_vals = joker2.sample_prior(8, return_logprobs=True)
        assert np.isfinite(ln_vals).all()