def test_call_block(): mat1 = np.random.rand(ORDER, ORDER) block_size = 16 row = 3 col = 5 print (p.row_block(mat1, row, block_size) == mat1[row*block_size:(row+1)*block_size]).all() print (p.col_block(mat1, col, block_size) == mat1[:, col*block_size:(col+1)*block_size]).all()
def test_process_dot(): q1 = Queue() mat1 = np.random.rand(ORDER, ORDER) row = 3 col = 5 block_size = 64 p.process_dot( q1, mat1, (row, col), block_size ) row1, col1, value = q1.get() print (np.dot( p.row_block(mat1, row, block_size), p.col_block(mat1, col, block_size)) == value).all()