def test_trim_cols_larger(self):
mat = sparse.csc_matrix(numpy.array([[1, 2, 0], [0, 5, 0]]))
(mat_reduced, old_ids) = filter_too_small(mat, 0, 2)
#self.assertIsInstance(mat_reduced, sparse.csc_matrix)
self.assertEqual((2, 1), mat_reduced.shape)
self.assertEqual(2, mat_reduced[0, 0])
self.assertEqual(5, mat_reduced[1, 0])
self.assertEqual(0, len(old_ids))
def test_trim_cols_larger(self):
mat = sparse.csc_matrix(numpy.array([[1,2,0],[0,5,0]]))
(mat_reduced, old_ids) = filter_too_small(mat, 0, 2)
#self.assertIsInstance(mat_reduced, sparse.csc_matrix)
self.assertEqual((2,1), mat_reduced.shape)
self.assertEqual(2, mat_reduced[0,0])
self.assertEqual(5, mat_reduced[1,0])
self.assertEqual(0, len(old_ids))
def test_trim_rows_smaller(self):
mat = sparse.csc_matrix(numpy.array([[1, 0, 0], [0, 0, 1], [0, 2, 3]]))
(mat_reduced, old_ids) = filter_too_small(mat, 2, 0)
#self.assertIsInstance(mat_reduced, sparse.csc_matrix)
self.assertEqual((1, 3), mat_reduced.shape)
self.assertEqual(0, mat_reduced[0, 0])
self.assertEqual(2, mat_reduced[0, 1])
self.assertEqual(3, mat_reduced[0, 2])
self.assertDictEqual({0: 2}, old_ids)
def test_trim_rows_smaller(self):
mat = sparse.csc_matrix(numpy.array([[1,0,0],[0,0,1], [0,2,3]]))
(mat_reduced, old_ids) = filter_too_small(mat, 2, 0)
#self.assertIsInstance(mat_reduced, sparse.csc_matrix)
self.assertEqual((1,3), mat_reduced.shape)
self.assertEqual(0, mat_reduced[0,0])
self.assertEqual(2, mat_reduced[0,1])
self.assertEqual(3, mat_reduced[0,2])
self.assertDictEqual({ 0 : 2 }, old_ids)
def test_trim_combined(self):
mat = sparse.csc_matrix(numpy.array([[0, 1, 0], [1, 0, 0], [0, 0, 0]]))
(mat_reduced, old_ids) = filter_too_small(mat, 1, 1)
self.assertEqual((2, 2), mat_reduced.shape)
def test_trim_combined(self):
mat = sparse.csc_matrix(numpy.array([[0,1,0],[1,0,0],[0,0,0]]))
(mat_reduced, old_ids) = filter_too_small(mat, 1, 1)
self.assertEqual((2,2), mat_reduced.shape)
