code: def test_get_object_corruption_scores():
      model = TransE(k=7)
  
      ent_matrix = np.array(
          [[4, 4, 4, 4, 4, 4, 9], [90, 90, 90, 90, 90, 90, 90]]
      ).astype(np.float32)
      triples = [
          np.array(
              [[1, 1, 1, 1, 1, 1, 1], [10, 10, 10, 10, 10, 10, 10]]
          ).astype(np.float32),
          np.array(
              [
                  [13, 13, 13, 13, 13, 13, 13],
                  [100, 100, 100, 100, 100, 100, 100],
              ]
          ).astype(np.float32),
          np.array(
              [[4, 4, 4, 4, 4, 4, 9], [90, 90, 90, 90, 90, 90, 90]]
          ).astype(np.float32),
      ]
      scores = np.around(
          model._get_object_corruption_scores(
              triples, ent_matrix
          ).numpy(),
          2,
      )
      assert (
          np.diag(scores) == np.array([-65.0, -140.0], dtype=np.float32)
      ).all(), "TransE: Scores don't match!"
  
imports: from ampligraph.latent_features.layers.scoring import TransE
  import pytest
  import tensorflow
  import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: test_get_object_corruption_scores() without return types