def METHOD_NAME(self, dataset: VHR10) -> None:
    for i in range(2):
        x = dataset[i]
        assert isinstance(x, dict)
        assert isinstance(x["image"], torch.Tensor)
        if dataset.split == "positive":
            assert isinstance(x["labels"], torch.Tensor)
            assert isinstance(x["boxes"], torch.Tensor)
            if "masks" in x:
                assert isinstance(x["masks"], torch.Tensor)