code: def test_get_tensor_size(graph):
      F, G, N = graph
      cugraph_store = CuGraphStore(F, G, N)
  
      tensor_attrs = cugraph_store.get_all_tensor_attrs()
      for tensor_attr in tensor_attrs:
          sz = N[tensor_attr.group_name]
  
          tensor_attr.index = np.arange(sz)
          assert cugraph_store.get_tensor_size(
              tensor_attr
          ) == torch.Size((sz,))
  
imports: import cugraph
  from cugraph_pyg.data.cugraph_store import CuGraphTensorAttr, CuGraphEdgeAttr, EdgeLayout
  from cugraph_pyg.data import CuGraphStore
  import cudf
  import cupy
  import numpy
  from cugraph.utilities.utils import import_optional, MissingModule
  import pytest
label_desc: Use image node_id 18 for calling a global function with example usage: test_get_tensor_size(graph) without return types