Source

pplp / benchmark / test_costflow.py

from graph import Graph
from costflow import costflow, costflow_cvxopt

def test_basic():
    g = Graph()
    g.add_arc(1, -1)
    g.add_arc(2, -1)
    g.add_arc(1, -1)
    g.add_node([2], [0])
    g.add_node([0], [1])
    g.add_node([1], [2])
    assert costflow(g) == -3
    assert costflow_cvxopt(g) == -3 

def test_simple1():
    g = Graph()    
    g.add_arc(1, 0)
    g.add_arc(1, 0)
    g.add_arc(1, -1)
    g.add_arc(1, -1)
    g.add_arc(10, 0)
    g.add_node([4], [0, 1])
    g.add_node([0], [2])
    g.add_node([1], [3])
    g.add_node([2, 3], [4])
    assert costflow(g) == -2
    assert costflow_cvxopt(g) == -2
    from cPickle import dump
    dump(g, open('graph.pi', 'w'), -1)
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.