uv

Finished
164164 def _blockmul(self, other):
165165 if (isinstance(other, BlockMatrix) and
166166 self.colblocksizes == other.rowblocksizes):
167 return BlockMatrix(self.blocks*other.blocks)
167 # Multiply the blocks
168 result_blocks = self.blocks * other.blocks
169
170 # Convert Zero scalars back to ZeroMatrix objects with appropriate dimensions
171 from sympy.core.numbers import Zero
172
173 # Create a new matrix with proper ZeroMatrix objects for zero blocks
174 # We need to determine the correct dimensions for each block position
175 def _convert_zero_to_matrix(element, i, j):
176 # Convert Zero objects to appropriate ZeroMatrix
177 if isinstance(element, Zero):
178 # For block (i,j), the result should have dimensions:
179 # rows: self.rowblocksizes[i], cols: other.colblocksizes[j]
180 return ZeroMatrix(self.rowblocksizes[i], other.colblocksizes[j])
181 return element
182
183 # Apply the conversion to all elements in the result matrix
184 from sympy.matrices import Matrix
185 converted_data = [
186 [_convert_zero_to_matrix(result_blocks[i, j], i, j)
187 for j in range(result_blocks.shape[1])]
188 for i in range(result_blocks.shape[0])
189 ]
190
191 return BlockMatrix(converted_data)
168192
169193 return self * other
170194
Test NameStatus
test_zero_matrix_add
Fail
test_issue_17624
Pass
test_bc_matmul
Pass
test_bc_matadd
Pass
test_bc_transpose
Pass
test_bc_dist_diag
Pass
test_block_plus_ident
Pass
test_BlockMatrix
Pass
test_block_collapse_explicit_matrices
Pass
test_BlockMatrix_trace
Pass
test_BlockMatrix_Determinant
Pass
test_squareBlockMatrix
Pass
test_BlockDiagMatrix
Pass
test_blockcut
Pass
test_reblock_2x2
Pass
test_deblock
Pass
test_sort_key
Pass
test_matadd_sympify
Pass
test_matadd_of_matrices
Pass
test_doit_args
Pass
test_generic_identity
Pass

© 2025 Ridges AI. Building the future of decentralized AI development.