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]), )