class_docstr: Tests of FullBatchNodeGenerator class
class_docstr_tok: ["Tests","of","FullBatchNodeGenerator","class"]
code: def test_fullbatch_generator_transform(self):
      G, _ = create_graph_features(is_directed=True)
  
      def func(features, A, **kwargs):
          return features, A.dot(A)
  
      generator = RelationalFullBatchNodeGenerator(
          G, "test", transform=func
      )
      assert generator.name == "test"
  
      As = []
      edge_types = sorted(
          set(e[-1] for e in G.edges(include_edge_type=True))
      )
      node_list = list(G.nodes())
      node_index = dict(zip(node_list, range(len(node_list))))
      for edge_type in edge_types:
          col_index = [
              node_index[n1]
              for n1, n2, etype in G.edges(include_edge_type=True)
              if etype == edge_type
          ]
          row_index = [
              node_index[n2]
              for n1, n2, etype in G.edges(include_edge_type=True)
              if etype == edge_type
          ]
          data = np.ones(len(col_index), np.float64)
  
          A = sps.coo_matrix(
              (data, (row_index, col_index)),
              shape=(len(node_list), len(node_list)),
          )
  
          As.append(A)
  
      for A_1, A_2 in zip(As, generator.As):
          np.testing.assert_array_equal(
              A_1.dot(A_1).todense(), A_2.todense()
          )
  
imports: from stellargraph.core.graph import *
  from stellargraph.mapper.full_batch_generators import RelationalFullBatchNodeGenerator
  import networkx
  import numpy
  import pytest
  import pandas
  import scipy.sparse
  from ..test_utils.graphs import relational_create_graph_features, example_hin_1
label_desc: Use image node_id 11 for calling the Test_RelationalFullBatchNodeGenerator obj's underlying member method code with example usage: obj.test_fullbatch_generator_transform() and returns: features,  A