Ejemplo n.º 1
0
    def discover_vendors(cls):
        """
        Discover GPU vendors of current node.

        :return: GPU vendor ID list.
        """
        return utils.discover_vendors()
Ejemplo n.º 2
0
 def test_discover_vendors(self, mock_popen):
     mock_popen.return_value = self.p
     gpu_venders = utils.discover_vendors()
     self.assertEqual(1, len(gpu_venders))
Ejemplo n.º 3
0
 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))
Ejemplo n.º 4
0
 def discover_vendors(cls):
     return utils.discover_vendors()