code: def test_should_return_random_gpu_ID_if_available(
      self, gpu_check, get_gpu_count
  ):
      gpu_check.return_value = True
      get_gpu_count.return_value = 1
      context = Context([0, 1, 2])
  
      selected_device = context.gpu_device()
      self.assertEqual(selected_device, 0)
  
file_path: evadb/test/unit_tests/executor/test_execution_context.py
imports: import unittest
  from mock import patch
  from evadb.constants import NO_GPU
  from evadb.executor.execution_context import Context
label_desc: Use image node_id 6 for calling the ExecutionContextTest obj's underlying member method code with example usage: obj.test_should_return_random_gpu_ID_if_available(gpu_check, get_gpu_count) without return types