code: def _check_get_arch(self, device_cc, expected_arch):
      with mock.patch("cupy.cuda.device.Device") as device_class:
          device_class.return_value.compute_capability = device_cc
          assert compiler._get_arch() == expected_arch
      cupy.clear_memo()
  
imports: import pickle
  import unittest
  from unittest import mock
  import cupy
  from cupy.cuda import compiler
label_desc: Use image node_id 2 for calling the TestNvrtcArch obj's underlying member method code with example usage: obj._check_get_arch(device_cc, expected_arch) without return types