def get_gpu_resources():
if not polyaxon_gpustat.has_gpu_nvidia:
return
try:
return polyaxon_gpustat.query()
except:
polyaxon_gpustat.has_gpu_nvidia = False
def test_query_mocked_nonexistent_pid(self, N, Process):
"""
Test a case where nvidia query returns non-existent pids (see #16, #18)
"""
_configure_mock(N, Process, scenario_nonexistent_pid=True)
gpustats = polyaxon_gpustat.query()
self.assertEqual(gpustats, MOCK_EXPECTED_OUTPUT_FULL)
def test_query_mocked(self, N, Process):
"""
A basic functionality test, in a case where everything is just normal.
"""
_configure_mock(N, Process)
gpustats = polyaxon_gpustat.query()
self.assertEqual(gpustats[0], MOCK_EXPECTED_OUTPUT_FULL[0])
self.assertEqual(gpustats[1], MOCK_EXPECTED_OUTPUT_FULL[1])
self.assertEqual(gpustats[2].pop('processes'), None)
mock_stats = copy.copy(MOCK_EXPECTED_OUTPUT_FULL[2])
mock_stats.pop('processes')
self.assertEqual(gpustats[2], mock_stats)
def get_gpu_resources():
try:
return polyaxon_gpustat.query()
except: # noqa
return []
def get_gpu_resources():
try:
return polyaxon_gpustat.query()
except: # noqa
return []
