Beispiel #1
0
def test_2():
    C = np.array([[1,1,0],
                  [0,1,1],
                  [0,1,1]])
    tC, m = _strongly_connected_subgraph(np.array(C))
    np.testing.assert_array_equal(tC, np.array([[1,1], [1,1]]))
    assert m == {1: 0, 2: 1}
Beispiel #2
0
def test_7():
    tC, m = _strongly_connected_subgraph(np.eye(3, k=1))
    assert tC.shape == (0, 0)
    assert m == {}
Beispiel #3
0
def test_6():
    tC, m = _strongly_connected_subgraph(np.eye(3), -1)
    print(tC)
Beispiel #4
0
def test_5():
    tC, m = _strongly_connected_subgraph(np.eye(3))
    assert tC.shape == (1,1)
Beispiel #5
0
def test_4():
    tC, m = _strongly_connected_subgraph(np.ones((3,3)))
    np.testing.assert_array_almost_equal(tC, np.ones((3,3)))
    assert m == {0:0, 1:1, 2:2}
Beispiel #6
0
def test_3():
    _strongly_connected_subgraph(np.zeros((3,3)))
Beispiel #7
0
def test_0():
    # what do you do with 1 state that is not even connected
    # to itself?
    tC, m = _strongly_connected_subgraph(np.zeros((1,1)))
    assert tC.shape == (0, 0)
    assert m == {}
Beispiel #8
0
def test_01():
    # but if that state does have a self-connection, it should be retained
    tC, m = _strongly_connected_subgraph(np.ones((1,1)))
    assert tC.shape == (1, 1)
    assert m == {0: 0}