from nose.tools import assert_true, assert_equal,assert_raises
import networkx as nx

def test_stochastic():
    G=nx.DiGraph()
    G.add_edge(0,1)
    G.add_edge(0,2)
    S=nx.stochastic_graph(G)
    assert_true(nx.is_isomorphic(G,S))
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])
    S=nx.stochastic_graph(G,copy=True)
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])


def test_stochastic_error():
    G=nx.Graph()
    assert_raises(Exception,nx.stochastic_graph,G)
    G=nx.MultiGraph()
    assert_raises(Exception,nx.stochastic_graph,G)

from nose.tools import assert_true, assert_equal,assert_raises
import networkx as nx

def test_stochastic():
    G=nx.DiGraph()
    G.add_edge(0,1)
    G.add_edge(0,2)
    S=nx.stochastic_graph(G)
    assert_true(nx.is_isomorphic(G,S))
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])
    S=nx.stochastic_graph(G,copy=True)
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])

def test_stochastic_ints():
    G=nx.DiGraph()
    G.add_edge(0,1,weight=1)
    G.add_edge(0,2,weight=1)
    S=nx.stochastic_graph(G)
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])



def test_stochastic_error():
    G=nx.Graph()
    assert_raises(Exception,nx.stochastic_graph,G)
    G=nx.MultiGraph()
    assert_raises(Exception,nx.stochastic_graph,G)

Test stochstic graph with ints
Test stochstic graph with ints

  from nose.tools import assert_true, assert_equal,assert_raises
  import networkx as nx
  
  def test_stochastic():
      G=nx.DiGraph()
      G.add_edge(0,1)
      G.add_edge(0,2)
      S=nx.stochastic_graph(G)
      assert_true(nx.is_isomorphic(G,S))
      assert_equal(sorted(S.edges(data=True)),
                   [(0, 1, {'weight': 0.5}), 
                    (0, 2, {'weight': 0.5})])
      S=nx.stochastic_graph(G,copy=True)
      assert_equal(sorted(S.edges(data=True)),
                   [(0, 1, {'weight': 0.5}), 
                    (0, 2, {'weight': 0.5})])
  
+ def test_stochastic_ints():
+     G=nx.DiGraph()
+     G.add_edge(0,1,weight=1)
+     G.add_edge(0,2,weight=1)
+     S=nx.stochastic_graph(G)
+     assert_equal(sorted(S.edges(data=True)),
+                  [(0, 1, {'weight': 0.5}), 
+                   (0, 2, {'weight': 0.5})])
+ 
+ 
  
  def test_stochastic_error():
      G=nx.Graph()
      assert_raises(Exception,nx.stochastic_graph,G)
      G=nx.MultiGraph()
      assert_raises(Exception,nx.stochastic_graph,G)
  
Test stochstic graph with ints
## Code Before:
from nose.tools import assert_true, assert_equal,assert_raises
import networkx as nx

def test_stochastic():
    G=nx.DiGraph()
    G.add_edge(0,1)
    G.add_edge(0,2)
    S=nx.stochastic_graph(G)
    assert_true(nx.is_isomorphic(G,S))
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])
    S=nx.stochastic_graph(G,copy=True)
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])


def test_stochastic_error():
    G=nx.Graph()
    assert_raises(Exception,nx.stochastic_graph,G)
    G=nx.MultiGraph()
    assert_raises(Exception,nx.stochastic_graph,G)

## Instruction:
Test stochstic graph with ints
## Code After:
from nose.tools import assert_true, assert_equal,assert_raises
import networkx as nx

def test_stochastic():
    G=nx.DiGraph()
    G.add_edge(0,1)
    G.add_edge(0,2)
    S=nx.stochastic_graph(G)
    assert_true(nx.is_isomorphic(G,S))
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])
    S=nx.stochastic_graph(G,copy=True)
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])

def test_stochastic_ints():
    G=nx.DiGraph()
    G.add_edge(0,1,weight=1)
    G.add_edge(0,2,weight=1)
    S=nx.stochastic_graph(G)
    assert_equal(sorted(S.edges(data=True)),
                 [(0, 1, {'weight': 0.5}), 
                  (0, 2, {'weight': 0.5})])



def test_stochastic_error():
    G=nx.Graph()
    assert_raises(Exception,nx.stochastic_graph,G)
    G=nx.MultiGraph()
    assert_raises(Exception,nx.stochastic_graph,G)
