code: def test_graph_schema_sampling_layout_1(example_graph_schema):
      # Create dummy graph schema
      schema = example_graph_schema(aa=0, bb=0)
  
      sampling_layout = schema.sampling_layout(["A"], [2, 2])
  
      assert len(sampling_layout) == 1
  
      assert sampling_layout[0][2] == ("A", [2, 3])
      assert sampling_layout[0][1] == ("B", [1])
      assert sampling_layout[0][0] == ("A", [0])
  
      # Check error handling
      schema = example_graph_schema(ab=2, ba=2, bb=0)
  
      # Only accepts lists & tuples
      with pytest.raises(TypeError):
          schema.sampling_layout("A", [1, 2])
  
      # Multiple edge types
      schema = example_graph_schema(ab=2, ba=2, bb=0)
      sampling_layout = schema.sampling_layout(("A",), [1, 2])
  
      assert len(sampling_layout) == 1
  
      assert sampling_layout[0] == [
          ("A", [0]),
          ("A", [1]),
          ("B", [2]),
          ("B", [3]),
          ("A", [4]),
          ("B", [5]),
          ("B", [6]),
          ("A", [7]),
          ("A", [8]),
          ("A", [9]),
          ("A", [10]),
      ]
  
file_path: stellargraph/tests/core/test_schema.py
imports: import pytest
  from stellargraph.core.schema import *
  import itertools
label_desc: Use image node_id 4 for calling a global function with example usage: test_graph_schema_sampling_layout_1(example_graph_schema) without return types