class_docstr: Tests for Weisfeiler-Lehman features.
class_docstr_tok: ["Tests","for","Weisfeiler-Lehman","features","."]
docstr: Prepare for test.
docstr_tok: ["Prepare","for","test","."]
code: def setUp(self) -> None:
      """Prepare for test."""
      dataset = Nations()
      self.initializer = pykeen.nn.init.WeisfeilerLehmanInitializer(
          triples_factory=dataset.training, shape=self.shape
      )
      self.num_entities = dataset.num_entities
  
imports: from typing import ClassVar
  import torch
  from class_resolver import HintOrType
  import pykeen.nn.init
  from pykeen.datasets import Nations
  from pykeen.nn.modules import ComplExInteraction, Interaction, QuatEInteraction
  from tests import cases
  from ..utils import needs_packages
label_desc: Use image node_id 1 for calling the WeisfeilerLehmanInitializerTestCase obj's underlying member method code with example usage: obj.setUp() without return types