コード例 #1
0
    def discover_vendors(cls):
        """
        Discover GPU vendors of current node.

        :return: GPU vendor ID list.
        """
        return utils.discover_vendors()
コード例 #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))
コード例 #3
0
ファイル: test_utils.py プロジェクト: joykechen/cyborg
 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))
コード例 #4
0
 def discover_vendors(cls):
     return utils.discover_vendors()