code: def test_embed_out_basic(self):
      for name, g in self.graphs:
          g = g.embed("rel", embedding_dim=self.d, **self.kwargs)
          num_nodes = len(set(g._edges["src"] + g._edges["dst"]))
          logging.debug("name: %s basic tests", name)
          self.assertEqual(g._edges.shape, self.edf.shape)
          self.assertEqual(
              set(g._edges[g._relation]), set(g._edges["rel"])
          )
          self.assertEqual(g._kg_embeddings.shape, (num_nodes, self.d))
  
imports: import os
  import pytest
  import pandas
  import unittest
  import graphistry
  import numpy
  from graphistry.embed_utils import lazy_embed_import_dep, check_cudf
  import logging
label_desc: Use image node_id 2 for calling the TestEmbed obj's underlying member method code with example usage: obj.test_embed_out_basic() without return types