code: def test_WatchYourStep_embeddings(barbell):
      generator = AdjacencyPowerGenerator(barbell, num_powers=5)
      wys = WatchYourStep(generator, embeddings_initializer="ones")
      x_in, x_out = wys.in_out_tensors()
  
      model = Model(inputs=x_in, outputs=x_out)
      model.compile(optimizer="adam", loss=graph_log_likelihood)
      embs = wys.embeddings()
  
      assert (embs == 1).all()
  
file_path: stellargraph/tests/layer/test_watch_your_step.py
imports: from stellargraph.layer import AttentiveWalk, WatchYourStep
  import numpy
  from ..test_utils.graphs import barbell
  from stellargraph.mapper import AdjacencyPowerGenerator
  from stellargraph.losses import graph_log_likelihood
  import pytest
  from tensorflow.keras import Model
  from ..None import test_utils
label_desc: Use image node_id 6 for calling a global function with example usage: test_WatchYourStep_embeddings(barbell) without return types