code: def test_graph_store():
      graph_store = MyGraphStore()
  
      assert str(graph_store) == "MyGraphStore()"
  
      coo = torch.tensor([0, 1]), torch.tensor([1, 2])
      csr = torch.tensor([0, 1, 2]), torch.tensor([1, 2])
      csc = torch.tensor([0, 1]), torch.tensor([0, 0, 1, 2])
  
      graph_store["edge_type", "coo"] = coo
      graph_store["edge_type", "csr"] = csr
      graph_store["edge_type", "csc"] = csc
  
      assert torch.equal(graph_store["edge_type", "coo"][0], coo[0])
      assert torch.equal(graph_store["edge_type", "coo"][1], coo[1])
      assert torch.equal(graph_store["edge_type", "csr"][0], csr[0])
      assert torch.equal(graph_store["edge_type", "csr"][1], csr[1])
      assert torch.equal(graph_store["edge_type", "csc"][0], csc[0])
      assert torch.equal(graph_store["edge_type", "csc"][1], csc[1])
  
      assert len(graph_store.get_all_edge_attrs()) == 3
  
      del graph_store["edge_type", "coo"]
      with pytest.raises(KeyError):
          graph_store["edge_type", "coo"]
  
      with pytest.raises(KeyError):
          graph_store["edge_type_2", "coo"]
  
imports: import pytest
  import torch
  from torch_geometric.data.graph_store import EdgeAttr, EdgeLayout
  from torch_geometric.testing import MyGraphStore, get_random_edge_index, withPackage
  from torch_geometric.utils import to_torch_coo_tensor, to_torch_csc_tensor, to_torch_csr_tensor
label_desc: Use image node_id 1 for calling a global function with example usage: test_graph_store() without return types