def discover_vendors(cls):
"""
Discover GPU vendors of current node.
:return: GPU vendor ID list.
"""
return utils.discover_vendors()
def test_discover_vendors(self, mock_popen):
mock_popen.return_value = self.p
gpu_venders = utils.discover_vendors()
self.assertEqual(1, len(gpu_venders))
def test_discover_vendors(self, mock_devices):
mock_devices.return_value = self.p.stdout.readlines()
gpu_vendors = utils.discover_vendors()
self.assertEqual(1, len(gpu_vendors))
def discover_vendors(cls):
return utils.discover_vendors()