Method test_default_names

tests/transforms/test_copy_itemsd.py:45–49

(self)

        np.testing.assert_allclose(result["img"], np.array([[0, 1], [1, 2]]))

    def test_default_names(self):
        input_data = {"img": np.array([[0, 1], [1, 2]]), "seg": np.array([[3, 4], [4, 5]])}
        result = CopyItemsd(keys=["img", "seg"], times=2, names=None)(input_data)
        for name in ["img_0", "seg_0", "img_1", "seg_1"]:
            self.assertTrue(name in result)

    def test_tensor_values(self):
        device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu:0")