def test_make_zero_matrix(self): m1 = [[1 for _ in range(4)] for _ in range(3)] m2 = util.make_zero_matrix_of_same_dimension(m1) self.assertEqual(len(m2), 3) for a in m2: self.assertEqual(len(a), 4) for b in a: self.assertEqual(b, 0)
def add_to(first, second): if first is None: first = util.make_zero_matrix_of_same_dimension(second) util.add_to(first, second) return first