Skip to content

Commit 638224c

Browse files
CEQ151CEQ151
authored andcommitted
test(image-watermark): update smoke for fingerprint payload
1 parent 9584576 commit 638224c

1 file changed

Lines changed: 21 additions & 2 deletions

File tree

.github/workflows/python-engine.yml

Lines changed: 21 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ jobs:
3131
dep = check_dependencies()
3232
assert dep['ok'], f'Missing deps: {dep}'
3333
34-
img = np.random.randint(0, 256, (800, 800, 3), dtype=np.uint8)
34+
img = np.random.default_rng(7).integers(0, 256, (800, 800, 3), dtype=np.uint8)
3535
3636
embedded = embed_watermark(
3737
img,
@@ -49,7 +49,26 @@ jobs:
4949
quality='balanced',
5050
)
5151
assert extracted['ok'], f'Extract failed: {extracted}'
52-
assert extracted['wm'] == 'hello world', f'Roundtrip failed: {extracted}'
52+
assert extracted['wm'].startswith('fp:'), f'Fingerprint roundtrip failed: {extracted}'
53+
assert extracted.get('payloadMode') == 'fingerprint64', extracted
54+
55+
text16_embedded = embed_watermark(
56+
img,
57+
'hello world',
58+
password=42,
59+
quality='balanced',
60+
payload_mode='text16',
61+
)
62+
assert text16_embedded['ok'], f'Text16 embed failed: {text16_embedded}'
63+
64+
text16_extracted = extract_watermark(
65+
text16_embedded['image'],
66+
password=42,
67+
quality='balanced',
68+
payload_mode='text16',
69+
)
70+
assert text16_extracted['ok'], f'Text16 extract failed: {text16_extracted}'
71+
assert text16_extracted['wm'] == 'hello world', f'Text16 roundtrip failed: {text16_extracted}'
5372
5473
print('All tests passed.')
5574

0 commit comments

Comments
 (0)