def setUp(self):
self.epg = EdgePairGraph(PAIRS)
def setUp(self): self.epg = EdgePairGraph(PAIRS)
class TestEdgePairGraph(unittest.TestCase):
def setUp(self):
self.epg = EdgePairGraph(PAIRS)
def test_pairs(self):
self.assert_(self.epg.pairs(copy=False) is PAIRS)
P = self.epg.pairs(copy=True)
self.assert_(P is not PAIRS)
assert_array_equal(P, PAIRS)
# test the directed case
P = self.epg.pairs(directed=False)
assert_array_equal(P, [[0, 1], [0, 2], [1, 1], [1, 2], [3, 3]])
def test_matrix(self):
M = self.epg.matrix()
assert_array_equal(M.toarray(), ADJ)
M = self.epg.matrix('dense')
assert_array_equal(M, ADJ)
M = self.epg.matrix('csr')
self.assertEqual(M.format, 'csr')
assert_array_equal(M.toarray(), ADJ)
def test_self_edges(self):
self.epg.add_self_edges()
expected = self.epg.pairs()
# Ensure that calling it again does the right thing.
self.epg.add_self_edges()
assert_array_equal(self.epg.pairs(), expected)
def test_symmetrize(self):
# Check that copy=True doesn't change anything
self.epg.symmetrize(copy=True)
assert_array_equal(self.epg.matrix('dense'), ADJ)
class TestEdgePairGraph(unittest.TestCase):
def setUp(self):
self.epg = EdgePairGraph(PAIRS)
def test_pairs(self):
self.assert_(self.epg.pairs(copy=False) is PAIRS)
P = self.epg.pairs(copy=True)
self.assert_(P is not PAIRS)
assert_array_equal(P, PAIRS)
# test the directed case
P = self.epg.pairs(directed=False)
assert_array_equal(P, [[0,1],[0,2],[1,1],[1,2],[3,3]])
def test_matrix(self):
M = self.epg.matrix()
assert_array_equal(M.toarray(), ADJ)
M = self.epg.matrix(dense=True)
assert_array_equal(M, ADJ)
M = self.epg.matrix(csr=True)
self.assertEqual(M.format, 'csr')
assert_array_equal(M.toarray(), ADJ)
def test_self_edges(self):
self.epg.add_self_edges()
expected = self.epg.pairs()
# Ensure that calling it again does the right thing.
self.epg.add_self_edges()
assert_array_equal(self.epg.pairs(), expected)
def test_symmetrize(self):
# Check that copy=True doesn't change anything
self.epg.symmetrize(copy=True)
assert_array_equal(self.epg.matrix(dense=True), ADJ)
