code: def test_chaining(self):
      for name, g in self.graphs:
          logging.debug(
              "name: %s test changing embedding dim with feats" % name
          )
          g = g.embed(
              "rel", use_feat=True, embedding_dim=self.d, **self.kwargs
          )
          g2 = g.embed(
              "rel",
              use_feat=True,
              embedding_dim=2 * self.d,
              **self.kwargs,
          )
          self.assertNotEqual(
              g._kg_embeddings.shape, g2._kg_embeddings.shape
          )
  
      [g.reset_caches() for _, g in self.graphs]
      for name, g in self.graphs:
          logging.debug(
              "name: %s test changing embedding dim without feats", name
          )
          g = g.embed(
              "rel", use_feat=False, embedding_dim=self.d, **self.kwargs
          )
          g2 = g.embed(
              "rel",
              use_feat=False,
              embedding_dim=2 * self.d,
              **self.kwargs,
          )
          self.assertNotEqual(
              g._kg_embeddings.shape, g2._kg_embeddings.shape
          )
  
      [g.reset_caches() for _, g in self.graphs]
      for name, g in self.graphs:
          logging.debug("name: %s test relationship change", name)
          g = g.embed(
              relation="rel",
              use_feat=False,
              embedding_dim=self.d,
              **self.kwargs,
          )
          g2 = g.embed(
              relation="src",
              use_feat=False,
              embedding_dim=self.d,
              **self.kwargs,
          )
          self.assertEqual(
              g._kg_embeddings.shape, g2._kg_embeddings.shape
          )
          self.assertNotEqual(
              np.linalg.norm(g._kg_embeddings - g2._kg_embeddings), 0
          )
  
      [g.reset_caches() for _, g in self.graphs]
      for name, g in self.graphs:
          logging.debug("name: %s test relationship change", name)
          g = g.embed(
              relation="rel",
              use_feat=False,
              embedding_dim=self.d,
              **self.kwargs,
          )
          g2 = g.embed(
              relation="rel",
              use_feat=True,
              embedding_dim=self.d,
              **self.kwargs,
          )
          self.assertEqual(
              g._kg_embeddings.shape, g2._kg_embeddings.shape
          )
          self.assertNotEqual(
              np.linalg.norm(g._kg_embeddings - g2._kg_embeddings), 0
          )
  
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 5 for calling the TestEmbed obj's underlying member method code with example usage: obj.test_chaining() without return types