code: def test_get_edges_count(reporter, partitioning):
      partitioning, args = partitioning
      edges_cnt = reporter.get_edges_count(partitioning)
      assert edges_cnt == [
          9,
          9,
      ], "Edges counts should be [9, 9] instead got {}.".format(
          edges_cnt
      )
  
file_path: AmpliGraph/tests/ampligraph/datasets/test_partitioning_reporter.py
imports: from ampligraph.datasets.partitioning_reporter import PartitioningReporter
  from ampligraph.datasets.graph_partitioner import NaiveGraphPartitioner, GraphDataLoader
  import pytest
  import mock
  from pytest_mock import mocker
  import pandas
  import numpy
label_desc: Use image node_id 13 for calling a global function with example usage: test_get_edges_count(reporter, partitioning) without return types