def test_is_almost_symmetric(self):
np.random.seed(8888)
vec1 = np.random.normal(0, 1, (100, 100))
vec2 = np.random.normal(0, 1, (100, 100))
corr = np.corrcoef(vec1, vec2)
self.assertTrue(gus.is_almost_symmetric(corr, atol=1e-15))
self.assertFalse(gus.is_symmetric(corr))
def test_undirected(self):
# with list index
indices = [0, -1, 1]
for idx in [indices, np.array(indices)]:
A, a = gus.remove_vertices(self.undirected,
idx,
return_removed=True)
self.assertIsInstance(a, np.ndarray)
assert_equal(A, np.array([[0, 18], [18, 0]]))
assert_equal(a, np.array([[11, 16], [23, 24], [12, 17]]))
self.assertTrue(gus.is_almost_symmetric(A))
# with integer index
indices = 0
A, a = gus.remove_vertices(self.undirected,
indices,
return_removed=True)
assert_equal(A, self.undirected[1:, 1:])
assert_equal(a, np.array([6, 11, 16, 21]))
self.assertTrue(gus.is_almost_symmetric(A))
assert_equal(
gus.remove_vertices(self.undirected, 0),
gus.remove_vertices(self.undirected, [0]),
)