code: def test_get_device(gpu_id, use_chainerx, expected):
      device = testing.get_device(gpu_id, use_chainerx)
      assert device.name == expected
  
imports: import pytest
  from chainermn import testing
  import chainer.testing.attr
label_desc: Use image node_id 2 for calling a global function with example usage: test_get_device(gpu_id, use_chainerx, expected) without return types