Skip to content

Commit 5f9129a

Browse files
test: assert hash_check has a weight set in functional test
1 parent 176ddc1 commit 5f9129a

1 file changed

Lines changed: 11 additions & 0 deletions

File tree

tests/functional/test_classify_single.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,3 +36,14 @@ async def test_classify_single(
3636
if result.error.code:
3737
msg = f"Image Result Error: {result.error.message}"
3838
pytest.fail(msg)
39+
40+
hash_checks = [
41+
classification
42+
for classification in result.classifications
43+
if "hash" in classification.label.lower()
44+
]
45+
46+
assert len(hash_checks) > 0, "No hash checks found in classifications"
47+
48+
for hash_check in hash_checks:
49+
assert hash_check.weight == 0.0

0 commit comments

Comments
 (0)