Python buildGraphの例

コード例 #1

ファイル: jetstreamstest.py プロジェクト: huynt1979/python

 def test_buildGraph(self):
     G = jetstreams.buildGraph(self.G, 0, 24)
     for e in G:
         print e
     
     self.assertEqual(G[0], (0,0,0))
     self.assertEqual(G[-1], (24,24,0))
     self.assertEqual(G[1], (0,5,10))
     self.assertEqual(G[-2], (21,22,2))

コード例 #2

ファイル: jetstreamstest.py プロジェクト: huynt1979/python

    def test_dag(self):
        G = jetstreams.buildGraph(self.G, 0, 24)        
        pred, cost, start, end = jetstreams.dag(G, self.constEnergy)

        t = end
        while True:
            print t,'->', pred[t]
            t = pred[t]
            if t == start:
                break
        
        self.assertEqual(cost, 352)