code: def test_embedding_dim(barbell, num_samples):
      scales = (0.1, 2, 3, 4)
      generator = GraphWaveGenerator(barbell, scales=scales, degree=10)
  
      sample_points = np.linspace(0, 1, num_samples)
  
      expected_embed_dim = len(sample_points) * len(scales) * 2
  
      for x in generator.flow(
          barbell.nodes(),
          sample_points=sample_points,
          batch_size=4,
          repeat=False,
      ):
          assert x.shape[1] == expected_embed_dim
  
file_path: stellargraph/tests/mapper/test_graphwave_generator.py
imports: from stellargraph.core.graph import *
  from stellargraph.mapper.graphwave_generator import GraphWaveGenerator, _empirical_characteristic_function
  from ..test_utils.graphs import barbell
  import numpy
  import pytest
  import scipy.sparse
  import tensorflow
label_desc: Use image node_id 9 for calling a global function with example usage: test_embedding_dim(barbell, num_samples) without return types