Hsyshasuzh/longmemory-geo / project /tests /model /test_dior_candidate_loading.py
Hsyshasuzh's picture
download
raw
828 Bytes
import unittest
from training.dior_reviewed_candidates import load_candidate
from training.detection_data import convert_detection_sample
class ReviewedDIORCandidateTests(unittest.TestCase):
def test_edge_candidates_are_kept_as_bridge_positives(self):
for identifier in ('dior_00716','dior_01118'):
with self.subTest(identifier=identifier):
sample=load_candidate(identifier)
example=convert_detection_sample(sample)
self.assertIn(2,example.class_ids.tolist())
self.assertEqual(example.image.shape,(1024,1024,3))
self.assertFalse(example.valid_mask[900,900])
def test_unknown_identifier_fails(self):
with self.assertRaises(ValueError):load_candidate('dior_not_in_queue')
if __name__=='__main__':unittest.main()

Xet Storage Details

Size:
828 Bytes
·
Xet hash:
7b276a2ab5d90817a9a575f67295c5e26524742621d0597ebe4a5d88d0c4805b

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.