test_Experiments.py 文件源码

def test_shot_plot(self):
        """tests the plotting function
        """

        init_prior = np.vectorize(lambda v: 2.56e9 / v**3)

        # Create the model and *true* EOS
        eos = EOSModel(init_prior)

        gun = Gun()

        data0 = gun({'eos': eos})
        old_dof = eos.get_c()
        old_dof[0] *= 1.02
        eos.update_dof(old_dof)
        data1 = gun({'eos': eos})

        gun.plot(level=3, data=[data0, data1])

        gun.plot(level=1, data=[data0, data1])

        plt.show()
# end