def testComputeRowSum_NumRowsWithRemainder(self):
m1 = csr_matrix([[1.0, 2.0, 0.0],
[0.0, 0.0, 3.0],
[4.0, 0.0, 5.0]])
# Process 1 process row 0, 1.
# Process 2 process row 2.
row_sum_dict = normalize.compute_row_sum_parallel(m1, 2)
self.assertEqual(row_sum_dict, {0: 3.0, 1: 3.0, 2: 9.0})
def testComputeRowSum_NumRowsWithMultipleOfParallelism(self):
m1 = csr_matrix([[1.0, 2.0, 0.0],
[0.0, 0.0, 3.0],
[4.0, 0.0, 5.0],
[6.0, 7.0, 8.0]])
# Process 1 process row 0, 1.
# Process 2 process row 2, 3.
row_sum_dict = normalize.compute_row_sum_parallel(m1, 2)
self.assertEqual(row_sum_dict, {0: 3.0, 1: 3.0, 2: 9.0, 3: 21.0})
