fix(ci): update watermark smoke workflow

Author: CEQ151

.github/workflows/python-engine.yml

@@ -13,24 +13,42 @@ jobs:
       run:
         working-directory: blind_watermark
     steps:
-      - uses: actions/checkout@34e114876b0b11c390a56381ad16ebd13914f8d5
-      - uses: actions/setup-python@a26af69be951a213d495a4c3e4e4022e16d87065
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
         with:
           python-version: '3.10'
       - name: Install dependencies
         run: |
           pip install -r requirements.txt
-      - name: Smoke test — import and roundtrip
+      - name: Smoke test - import and roundtrip
         shell: python {0}
         env:
           PYTHONPATH: .
         run: |
           from rwm_engine import embed_watermark, extract_watermark, check_dependencies
           import numpy as np
+
           dep = check_dependencies()
           assert dep['ok'], f'Missing deps: {dep}'
+
           img = np.random.randint(0, 256, (800, 800, 3), dtype=np.uint8)
-          wm = embed_watermark(img, 'hello world', password=42, quality='balanced')
-          extracted = extract_watermark(wm, password=42, quality='balanced')
-          assert extracted == 'hello world', f'Roundtrip failed: {extracted}'
-          print('All tests passed.')
+
+          embedded = embed_watermark(
+              img,
+              'hello world',
+              password=42,
+              quality='balanced',
+          )
+          assert embedded['ok'], f'Embed failed: {embedded}'
+          assert embedded['image'] is not None, f'Embed image missing: {embedded}'
+          assert embedded['engine_used'] in {'legacy', 'neural'}, embedded
+
+          extracted = extract_watermark(
+              embedded['image'],
+              password=42,
+              quality='balanced',
+          )
+          assert extracted['ok'], f'Extract failed: {extracted}'
+          assert extracted['wm'] == 'hello world', f'Roundtrip failed: {extracted}'
+
+          print('All tests passed.')