class TestFSVD(unittest.TestCase):
def setUp(self):
self.test_matrix = np.loadtxt(get_data_path('full_sym_matrix.txt'))
self.fsvd = FSVD()
def test_svd(self):
"""
Ensure eigenvalues and eigenvectors are returned.
"""
DIMENSIONS_OUT = 10
eigenvectors, eigenvalues = self.fsvd.run(self.test_matrix,
DIMENSIONS_OUT)
self.assertEqual(eigenvectors.shape, (100, DIMENSIONS_OUT))
self.assertEqual(eigenvalues.shape, (DIMENSIONS_OUT,))
def setUp(self):
self.test_matrix = np.loadtxt(get_data_path('full_sym_matrix.txt'))
self.fsvd = FSVD()
