Update test_attention_v1.py

Author: NSDie

tests/ut/attention/test_attention_v1.py

@@ -453,9 +453,9 @@ def test_forward_decode_only_swa(self, mock_fused_infer_attention_score,
                                        kv_cache,
                                        metadata,
                                        trace_flag=False)
-
+        print(output.shape)
         mock_fused_infer_attention_score.assert_called_once()
-        assert output.shape == (10, 8 * 64)
+        assert output.shape == (10, 8, 64)
 
     @patch('vllm_ascend.attention.attention_v1.is_310p', return_value=False)
     @patch('torch_npu._npu_reshape_and_cache')