def test_setitem(self): m = BlockMatrix(2, 2) m[0, 1] = self.block_m self.assertFalse(m.is_empty_block(0, 1)) self.assertEqual(m.row_block_sizes()[0], self.block_m.shape[0]) self.assertEqual(m.col_block_sizes()[1], self.block_m.shape[1]) self.assertEqual(m[0, 1].shape, self.block_m.shape)
def test_setitem(self): m = BlockMatrix(2, 2) m.set_block(0, 1, self.block_m) self.assertFalse(m.is_empty_block(0, 1)) self.assertEqual(m._brow_lengths[0], self.block_m.shape[0]) self.assertEqual(m._bcol_lengths[1], self.block_m.shape[1]) self.assertEqual(m.get_block(0, 1).shape, self.block_m.shape)
def test_setitem(self): m = BlockMatrix(2, 2) m[0, 1] = self.block_m self.assertFalse(m.is_empty_block(0, 1)) self.assertEqual(m.row_block_sizes()[0], self.block_m.shape[0]) self.assertEqual(m.col_block_sizes()[1], self.block_m.shape[1]) self.assertEqual(m[0, 1].shape, self.block_m.shape)