code: def test_dataset_call(self, data, labels):
      dataset = BaseDataset(data, labels)
  
      assert torch.all(dataset[0]["data"] == data[0])
      assert torch.all(dataset[1]["labels"] == labels[1])
      assert torch.all(dataset.data == data)
      assert torch.all(dataset.labels == labels)
  
      index = torch.randperm(1000)[:3]
  
      assert torch.all(dataset[index]["data"] == data[index])
      assert torch.all(dataset[index]["labels"] == labels[index])
  
file_path: benchmark_VAE/tests/test_datasets.py
imports: import os
  import pytest
  import torch
  from pythae.data.datasets import BaseDataset
label_desc: Use image node_id 1 for calling the Test_Dataset obj's underlying member method code with example usage: obj.test_dataset_call(data, labels) without return types