code: def test_load_wn18():
      wn18 = load_wn18()
      assert len(wn18["train"]) == 141442
      assert len(wn18["valid"]) == 5000
      assert len(wn18["test"]) == 5000
  
      ent_train = np.union1d(
          np.unique(wn18["train"][:, 0]), np.unique(wn18["train"][:, 2])
      )
      ent_valid = np.union1d(
          np.unique(wn18["valid"][:, 0]), np.unique(wn18["valid"][:, 2])
      )
      ent_test = np.union1d(
          np.unique(wn18["test"][:, 0]), np.unique(wn18["test"][:, 2])
      )
      distinct_ent = np.union1d(
          np.union1d(ent_train, ent_valid), ent_test
      )
      distinct_rel = np.union1d(
          np.union1d(
              np.unique(wn18["train"][:, 1]),
              np.unique(wn18["train"][:, 1]),
          ),
          np.unique(wn18["train"][:, 1]),
      )
  
      assert len(distinct_ent) == 40943
      assert len(distinct_rel) == 18
  
file_path: AmpliGraph/tests/ampligraph/datasets/test_datasets.py
imports: from ampligraph.datasets import load_wn18, load_fb15k, load_fb15k_237, load_yago3_10, load_wn18rr, load_wn11, load_fb13
  from ampligraph.datasets.datasets import _clean_data
  import os
  import numpy
  import pytest
label_desc: Use image node_id 2 for calling a global function with example usage: test_load_wn18() without return types