python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-08-03 12:00:10 +03:00
Code Issues Packages Projects Releases Wiki Activity

use deterministic hash function

Browse Source
This commit is contained in:
Yay295 2024-09-19 00:58:59 -05:00
parent 05e2161fbe
commit 77f42c52b9
1 changed files with 32 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
Tests/test_image_histogram.py
View File

@ -1,5 +1,8 @@
from __future__ import annotations from __future__ import annotations
import base64
import hashlib
import pytest import pytest
from PIL import Image from PIL import Image
@ -7,34 +10,38 @@ from PIL import Image
from .helper import hopper from .helper import hopper
expected_data = { expected_data = {
"1": (256, 0, 10994, -6588993366496339844), "1": (256, 0, 10994, b"zk4Al^)(jtioYSyJT@emR4<LAz}yHLJ6>*b*%R1K"),
"CMYK": (1024, 0, 16384, 6499941520588169132), "CMYK": (1024, 0, 16384, b"zqm8<<;r>h42v|$3xNIR#fG2=P&z(awe2&{GK(8o"),
"F": (256, 0, 662, -6473002676204311189), "F": (256, 0, 662, b"*EIrB8n_NEx9e()#ao<>L)@gEjm|N%I8B)YhA8V~"),
"HSV": (768, 0, 1696, -6543359421857819796), "HSV": (768, 0, 1696, b"F6vt%@sQf%X04Md4^n!Qpv!qJ8Oz()CqPx=rjFvu"),
"I": (256, 0, 662, -6473002676204311189), "I": (256, 0, 662, b"*EIrB8n_NEx9e()#ao<>L)@gEjm|N%I8B)YhA8V~"),
"I;16": (256, 0, 8192, 5889973619001565923), "I;16": (256, 0, 8192, b"S+c=3i+Fs3wK2>Q<8rq@PgsAg=nE1VLdMtHZ2K8$"),
"I;16B": (256, 0, 8192, 5889973619001565923), "I;16B": (256, 0, 8192, b"S+c=3i+Fs3wK2>Q<8rq@PgsAg=nE1VLdMtHZ2K8$"),
"I;16L": (256, 0, 8192, 5889973619001565923), "I;16L": (256, 0, 8192, b"S+c=3i+Fs3wK2>Q<8rq@PgsAg=nE1VLdMtHZ2K8$"),
"I;16N": (256, 0, 8192, 5889973619001565923), "I;16N": (256, 0, 8192, b"S+c=3i+Fs3wK2>Q<8rq@PgsAg=nE1VLdMtHZ2K8$"),
"L": (256, 0, 662, -2745750633816556827), "L": (256, 0, 662, b"EmZC)FNJ#AK=O?2(qxYeY#*-vk97Iz%f8!<LaY}1"),
"LA": (512, 0, 662, -7010327511106502095), "LA": (512, 0, 662, b"Eds{eGw)qbudo%xtTJ_6#d5In@<fQMK3Kn%5!9;0"),
"La": (512, 0, 662, -7010327511106502095), "La": (512, 0, 662, b"Eds{eGw)qbudo%xtTJ_6#d5In@<fQMK3Kn%5!9;0"),
"LAB": (768, 0, 1946, -5543905267928649909), "LAB": (768, 0, 1946, b"9X3PUSDkz%k~jXeBH}f4?u?ga;js`?-81}2(RAPF"),
"P": (256, 0, 1551, -8229466336626460515), "P": (256, 0, 1551, b"mCdCL$^NJhLi#i!x{jz~<&y%a{iVi(H(JP}W5-Op"),
"PA": (512, 0, 1551, 5750919599917307801), "PA": (512, 0, 1551, b"r_3_AMV`<vvmH_kY7ulvv|2QT==P{6rpFz6)D>~q"),
"RGB": (768, 4, 675, -7521681876442962674), "RGB": (768, 4, 675, b"!TjBr^o$}ZXlfh|aIU7&4VKP8=rBq&RecVDIcndZ"),
"RGBA": (1024, 0, 16384, -4362402434932696434), "RGBA": (1024, 0, 16384, b"du`JBXvzDrPu^Ybc}8Y+4y1MDTEK|!Q|rR~Jk^@J"),
"RGBa": (1024, 0, 16384, -4362402434932696434), "RGBa": (1024, 0, 16384, b"du`JBXvzDrPu^Ybc}8Y+4y1MDTEK|!Q|rR~Jk^@J"),
"RGBX": (1024, 0, 16384, -4362402434932696434), "RGBX": (1024, 0, 16384, b"du`JBXvzDrPu^Ybc}8Y+4y1MDTEK|!Q|rR~Jk^@J"),
"YCbCr": (768, 0, 1908, -1513926274659056238), "YCbCr": (768, 0, 1908, b"%z+JjEuI^YFOt*}($tSuSk^nX-~-HI>QDL>T%9H9"),
} }
# Python's basic hash() function isn't necessarily deterministic, so use this instead.
def deterministic_hash(data: list[int]) -> bytes:
data_bytes = str(data).encode("ascii")
hash_digest = hashlib.sha256(data_bytes).digest()
return base64.b85encode(hash_digest)
@pytest.mark.parametrize("mode", Image.MODES) @pytest.mark.parametrize("mode", Image.MODES)
def test_histogram(mode: str) -> None: def test_histogram(mode: str) -> None:
h = hopper(mode).histogram() h = hopper(mode).histogram()
# If all values are checked in the same assert and they don't match, data = (len(h), min(h), max(h), deterministic_hash(h))
# the error message gets truncated. So use multiple asserts to see assert data == expected_data[mode]
# the full message.
assert (len(h), min(h), max(h)) == expected_data[mode][:3]
assert hash(tuple(h)) == expected_data[mode][3]