Immune

Finished
22
33from sympy import ask, Q
44from sympy.core import Basic, Add
5from sympy.core.numbers import Zero
56from sympy.core.compatibility import range
67from sympy.strategies import typed, exhaust, condition, do_one, unpack
78from sympy.strategies.traverse import bottom_up
164165 def _blockmul(self, other):
165166 if (isinstance(other, BlockMatrix) and
166167 self.colblocksizes == other.rowblocksizes):
167 return BlockMatrix(self.blocks*other.blocks)
168 # Perform block multiplication
169 result_blocks = self.blocks * other.blocks
170
171 # Convert Zero scalar objects back to ZeroMatrix objects
172 # with appropriate dimensions
173 from sympy.matrices.immutable import ImmutableDenseMatrix
174 corrected_blocks = []
175 for i in range(result_blocks.shape[0]):
176 row = []
177 for j in range(result_blocks.shape[1]):
178 elem = result_blocks[i, j]
179 if isinstance(elem, Zero):
180 # Determine the dimensions for this ZeroMatrix
181 rows = self.rowblocksizes[i]
182 cols = other.colblocksizes[j]
183 row.append(ZeroMatrix(rows, cols))
184 else:
185 row.append(elem)
186 corrected_blocks.append(row)
187
188 return BlockMatrix(ImmutableDenseMatrix(corrected_blocks))
168189
169190 return self * other
170191
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.