BAAI/Brainmu-Spike

1import importlib.util,tempfile,unittest2from pathlib import Path3from PIL import Image4spec=importlib.util.spec_from_file_location("evaluation",Path(__file__).resolve().parents[1]/"scripts/infer.py")5m=importlib.util.module_from_spec(spec);spec.loader.exec_module(m)6class EvaluationTests(unittest.TestCase):7 def test_complete_pair_and_hashes(self):8  with tempfile.TemporaryDirectory() as d:9   p=Path(d);(p/"x.dat").write_bytes(bytes(512500));Image.new("RGB",(400,250)).save(p/"x.png")10   files,rows=m.validate_pairs(p,p,1)11   self.assertEqual(len(files),1);self.assertEqual(len(rows[0]["spike_sha256"]),64)12   with self.assertRaises(ValueError):m.validate_pairs(p,p,1000)13 def test_missing_or_bad_gt(self):14  with tempfile.TemporaryDirectory() as d:15   p=Path(d);(p/"x.dat").write_bytes(bytes(512500))16   with self.assertRaises(ValueError):m.validate_pairs(p,p,1)17   Image.new("RGB",(10,10)).save(p/"x.png")18   with self.assertRaises(ValueError):m.validate_pairs(p,p,1)19if __name__=="__main__":unittest.main()20