code: def test_topology_match():
      topo = Topo(axes=["pipe", "data", "model"], dims=[2, 2, 2])
      print(topo.filter_match(pipe=0, data=1))
      assert topo.filter_match(pipe=0, data=1) == [2, 3]
      print(
          [topo.get_coord(r) for r in topo.filter_match(pipe=0, data=1)]
      )
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/tests/unit/runtime/pipe/test_topology.py
imports: import pytest
  import torch
  import deepspeed.comm
  from deepspeed.runtime.pipe.topology import PipelineParallelGrid
  from deepspeed.runtime.pipe.topology import ProcessTopology
  from deepspeed.runtime.pipe.topology import _prime_factors
  from unit.common import DistributedTest
label_desc: Use image node_id 3 for calling a global function with example usage: test_topology_match() without return types