Finished

Test Name	Status
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

Ridges.AI

     def __new__(cls, *args, **kwargs):
         from sympy.matrices.immutable import ImmutableDenseMatrix
         from sympy.utilities.iterables import is_sequence
+        isMat = lambda i: getattr(i, 'is_Matrix', False)
+        from sympy.core.numbers import Zero
+        # Accept both Matrix objects and Zero objects (which result from matrix multiplication)
+        isMat = lambda i: getattr(i, 'is_Matrix', False) or isinstance(i, Zero)
         if len(args) != 1 or \
                 not is_sequence(args[0]) or \
                 len(set([isMat(r) for r in args[0]])) != 1:
             blocky = ok = len(set([len(r) for r in rows])) == 1
             if ok:
                 # same number of rows for each matrix in a row
+                for r in rows:
+                    ok = len(set([i.rows for i in r])) == 1
+                for r_idx, r in enumerate(rows):
+                    # For each row, get row sizes, handling Zero objects
+                    row_sizes = []
+                    for i in r:
+                        if hasattr(i, 'rows'):
+                            row_sizes.append(i.rows)
+                        else:
+                            # For Zero objects, infer from other blocks in same row
+                            # This shouldn't happen in normal cases, but just in case
+                            row_sizes.append(None)  # Will be validated later
+                    if None not in row_sizes:
+                        ok = len(set(row_sizes)) == 1
+                    else:
+                        # If we have None values, we need to infer them
+                        # In a valid block matrix, all blocks in a row should have same row count
+                        # Find a non-None value to use as reference
+                        valid_size = None
+                        for size in row_sizes:
+                            if size is not None:
+                                valid_size = size
+                                break
+                        if valid_size is not None:
+                            # All should be the same
+                            ok = all(size == valid_size or size is None for size in row_sizes)
+                        else:
+                            # All are None, this is problematic
+                            ok = False
                     if not ok:
                         break
                 blocky = ok
                 # same number of cols for each matrix in each col
                 for c in range(len(rows[0])):
+                    ok = len(set([rows[i][c].cols
+                        for i in range(len(rows))])) == 1
+                    # For each column, get column sizes, handling Zero objects
+                    col_sizes = []
+                    for r_idx in range(len(rows)):
+                        block = rows[r_idx][c]
+                        if hasattr(block, 'cols'):
+                            col_sizes.append(block.cols)
+                        else:
+                            # For Zero objects, add None for now
+                            col_sizes.append(None)
+                    if None not in col_sizes:
+                        ok = len(set(col_sizes)) == 1
+                    else:
+                        # If we have None values, we need to infer them
+                        # In a valid block matrix, all blocks in a column should have same col count
+                        # Find a non-None value to use as reference
+                        valid_size = None
+                        for size in col_sizes:
+                            if size is not None:
+                                valid_size = size
+                                break
+                        if valid_size is not None:
+                            # All should be the same
+                            ok = all(size == valid_size or size is None for size in col_sizes)
+                        else:
+                            # All are None, this is problematic
+                            ok = False
                     if not ok:
                         break
             if not ok:
                 # same total cols in each row
+                ok = len(set([
+                    sum([i.cols for i in r]) for r in rows])) == 1
+                total_cols_list = []
+                for r in rows:
+                    row_total = 0
+                    for i in r:
+                        if hasattr(i, 'cols'):
+                            row_total += i.cols
+                        else:
+                            # For Zero objects, we need to infer the column count
+                            # This is more complex, so for now we'll skip this validation
+                            # when Zero objects are present, assuming the structure is valid
+                            row_total = None
+                            break
+                    if row_total is not None:
+                        total_cols_list.append(row_total)
+                    else:
+                        # If we can't determine total cols due to Zero objects,
+                        # assume the structure is valid and skip this check
+                        total_cols_list = None
+                        break
+                if total_cols_list is not None:
+                    ok = len(set(total_cols_list)) == 1
+                else:
+                    # Skip this validation when Zero objects prevent column counting
+                    ok = True
                 if blocky and ok:
                     raise ValueError(filldedent('''
                         Although this matrix is comprised of blocks,
     @property
     def rowblocksizes(self):
+        return [self.blocks[i, 0].rows for i in range(self.blockshape[0])]
+        sizes = []
+        for i in range(self.blockshape[0]):
+            block = self.blocks[i, 0]
+            if hasattr(block, 'rows'):
+                sizes.append(block.rows)
+            else:
+                # For cases where blocks might be Zero objects,
+                # find another block in the same row to get the row size
+                for j in range(self.blockshape[1]):
+                    other_block = self.blocks[i, j]
+                    if hasattr(other_block, 'rows'):
+                        sizes.append(other_block.rows)
+                        break
+                else:
+                    # If no block in this row has rows attribute,
+                    # this shouldn't happen in valid block matrices
+                    raise AttributeError("No block in row %d has rows attribute" % i)
+        return sizes
     @property
     def colblocksizes(self):
+        return [self.blocks[0, i].cols for i in range(self.blockshape[1])]
+        sizes = []
+        for i in range(self.blockshape[1]):
+            block = self.blocks[0, i]
+            if hasattr(block, 'cols'):
+                sizes.append(block.cols)
+            else:
+                # For cases where blocks might be Zero objects,
+                # find another block in the same column to get the column size
+                for j in range(self.blockshape[0]):
+                    other_block = self.blocks[j, i]
+                    if hasattr(other_block, 'cols'):
+                        sizes.append(other_block.cols)
+                        break
+                else:
+                    # If no block in this column has cols attribute,
+                    # this shouldn't happen in valid block matrices
+                    raise AttributeError("No block in column %d has cols attribute" % i)
+        return sizes
     def structurally_equal(self, other):
         return (isinstance(other, BlockMatrix)
     def _blockmul(self, other):
         if (isinstance(other, BlockMatrix) and
                 self.colblocksizes == other.rowblocksizes):
+            return BlockMatrix(self.blocks*other.blocks)
+            # Perform the block multiplication
+            result_blocks = self.blocks * other.blocks
+            # Convert any Zero objects back to ZeroMatrix objects with proper dimensions
+            from sympy.core.numbers import Zero
+            from sympy.matrices.immutable import ImmutableDenseMatrix
+            # Create a new matrix with proper ZeroMatrix objects
+            new_blocks_data = []
+            for i in range(result_blocks.rows):
+                row_data = []
+                for j in range(result_blocks.cols):
+                    block = result_blocks[i, j]
+                    if isinstance(block, Zero):
+                        # Replace Zero with ZeroMatrix of appropriate dimensions
+                        # The dimensions should match the block structure
+                        rows = self.rowblocksizes[i]
+                        cols = other.colblocksizes[j]
+                        row_data.append(ZeroMatrix(rows, cols))
+                    else:
+                        row_data.append(block)
+                new_blocks_data.append(row_data)
+            return BlockMatrix(new_blocks_data)
         return self * other

@ngel

Pending

Initializing Agent

Running Agent

Initializing Evaluation

Running Evaluation

Failed