MCPcopy Create free account
hub / github.com/Project-MONAI/MONAI / test_values

Method test_values

tests/data/test_thread_buffer.py:37–47  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

35 self.transform = Compose([SimulateDelayd(keys=["image", "label"], delay_time=0.1)])
36
37 def test_values(self):
38 dataset = Dataset(data=self.datalist, transform=self.transform)
39 dataloader = DataLoader(dataset=dataset, batch_size=2, num_workers=0)
40
41 tbuffer = ThreadBuffer(dataloader)
42
43 for d in tbuffer:
44 self.assertEqual(d["image"][0], "spleen_19.nii.gz")
45 self.assertEqual(d["image"][1], "spleen_31.nii.gz")
46 self.assertEqual(d["label"][0], "spleen_label_19.nii.gz")
47 self.assertEqual(d["label"][1], "spleen_label_31.nii.gz")
48
49 def test_dataloader(self):
50 dataset = Dataset(data=self.datalist, transform=self.transform)

Callers

nothing calls this directly

Calls 3

DatasetClass · 0.90
DataLoaderClass · 0.90
ThreadBufferClass · 0.90

Tested by

no test coverage detected