simulation_test.py 文件源码

def test_from_Thetas(self):
        Theta1 = np.array([[1,  0, .3],
                           [0,  .5, 0],
                           [.3, 0,  1]])
        Theta2 = np.array([[1,  .3, 0],
                           [.3, .5, 0],
                           [0,  0,  1]])
        Thetas = np.zeros((10, 3, 3))
        for i in range(5):
            Thetas[i] = Theta1
        for i in range(5, 10):
            Thetas[i] = Theta2
        DGM = DynamicGraphicalModel.from_Thetas(Thetas)
        self.assertEqual(len(DGM.graphs), 2)
        G1, G2 = DGM.graphs
        self.assertEqual(G1.n_edges, 1)
        self.assertEqual(G2.n_edges, 1)
        self.assertTrue(np.allclose(G1.Theta, Theta1))
        self.assertTrue(np.allclose(G2.Theta, Theta2))