67 lines
4.3 KiB
Python
67 lines
4.3 KiB
Python
import sys
|
|
from pathlib import Path
|
|
sys.path.insert(0,str(Path(__file__).resolve().parents[1]/"tools"))
|
|
import copy, struct, unittest, tempfile,hashlib
|
|
from pathlib import Path
|
|
from PIL import Image
|
|
from hd_frame_policy import validate_frame_policy, _indices
|
|
|
|
class FramePolicyTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.frame={"logical_width":2,"logical_height":2}
|
|
self.raw=struct.pack("<iiiii",1,0,0,2,2)+bytes([5,239,0,30])
|
|
def test_absent_or_explicit_default_metadata_is_valid(self):
|
|
validate_frame_policy(self.frame,self.raw)
|
|
self.frame.update(preserve_reserved_coverage=False,native_nearest_cues=[])
|
|
validate_frame_policy(self.frame,self.raw)
|
|
def test_valid_explicit_flags_and_sorted_ordinary_cells(self):
|
|
self.frame.update(preserve_reserved_coverage=True,native_nearest_cues=[[0,0],[1,0]])
|
|
validate_frame_policy(self.frame,self.raw)
|
|
def test_coverage_type_is_strict(self):
|
|
for value in [0,1,"yes",None,[],{}]:
|
|
with self.subTest(value=value),self.assertRaises(ValueError):
|
|
validate_frame_policy(dict(self.frame,preserve_reserved_coverage=value))
|
|
def test_coordinate_structure_bounds_types_order_duplicates_limit(self):
|
|
for value in [None,"yes",{},[[0]],[[0,0,0]],[[True,0]],[[0.0,0]],[[-1,0]],[[2,0]],[[0,2]],[[0,0],[0,0]],[[1,0],[0,0]],[[0,0]]*65]:
|
|
with self.subTest(value=value),self.assertRaises(ValueError):
|
|
validate_frame_policy(dict(self.frame,native_nearest_cues=value),self.raw)
|
|
def test_ordinary_cues_never_use_reserved_source_indices(self):
|
|
self.frame['native_nearest_cues']=[[0,0]]
|
|
for label in [240,244,245,249,250,255]:
|
|
with self.subTest(label=label),self.assertRaises(ValueError):
|
|
validate_frame_policy(self.frame,self.raw[:20]+bytes([label,5,5,5]))
|
|
def test_sparse_source_requires_actual_complete_coverage(self):
|
|
header=struct.pack('<iiiii',0,0,0,2,2)
|
|
end=struct.pack('<iiii',0,0,-1,0)
|
|
sparse=header+struct.pack('<iiii',0,0,0,2)+bytes([5,5])+end
|
|
self.frame['native_nearest_cues']=[[0,0]]
|
|
with self.assertRaises(ValueError):validate_frame_policy(self.frame,sparse)
|
|
complete=header+struct.pack('<iiii',0,0,0,2)+bytes([5,5])+struct.pack('<iiii',0,1,0,2)+bytes([5,5])+end
|
|
validate_frame_policy(self.frame,complete)
|
|
def test_checked_source_rejects_invalid_or_truncated_original(self):
|
|
self.frame['native_nearest_cues']=[[0,0]]
|
|
for payload in [b'',self.raw[:22],struct.pack('<iiiii',1,0,0,4096,4096),struct.pack('<iiiii',3,0,0,2,2)]:
|
|
with self.subTest(payload=payload),self.assertRaises(ValueError):validate_frame_policy(self.frame,payload)
|
|
def test_matched_candidate_tool_checks_flags_and_exact_target_bindings(self):
|
|
from hd_candidate_review import proposal_frames
|
|
f=dict(self.frame,frame=0,file='a.png',origin=[0,0],source_sha256='00',native_nearest_cues=[[0,0]])
|
|
pack={'sequences':[{'name':'a','base_asset':1,'frame_count':1,'frames':[f]}]}
|
|
self.assertEqual(set(proposal_frames(pack)),{'00001'})
|
|
bad=copy.deepcopy(pack);bad['sequences'][0]['frames'][0]['preserve_reserved_coverage']=1
|
|
with self.assertRaises(ValueError):proposal_frames(bad)
|
|
bad=copy.deepcopy(pack);bad['targets']=[dict(f,asset='00001',native_nearest_cues=[])]
|
|
with self.assertRaises(ValueError):proposal_frames(bad)
|
|
def test_matched_art_audit_checks_original_index_and_transparency(self):
|
|
from hd_art_audit import validate_hd_frame
|
|
with tempfile.TemporaryDirectory() as temp:
|
|
root=Path(temp);Image.new('RGBA',(8,8),(10,20,30,255)).save(root/'a.png')
|
|
f=dict(self.frame,frame=0,file='a.png',origin=[0,0],source_sha256=hashlib.sha256(self.raw).hexdigest(),native_nearest_cues=[[0,0]])
|
|
picture=Image.new('RGBA',(2,2),(10,20,30,255))
|
|
self.assertEqual(validate_hd_frame(root,f,self.raw,picture),8*8*4)
|
|
reserved=self.raw[:20]+bytes([253,5,5,5]);f['source_sha256']=hashlib.sha256(reserved).hexdigest()
|
|
with self.assertRaisesRegex(ValueError,'ordinary'):validate_hd_frame(root,f,reserved,picture)
|
|
def test_unknown_frame_policy_is_rejected(self):
|
|
with self.assertRaisesRegex(ValueError,'unknown'):validate_frame_policy(dict(self.frame,imaginary_cue_rule=True))
|
|
|
|
if __name__=='__main__':unittest.main()
|