def test_get_gpu_info(self):
info = get_gpu_info()
try:
assert 'tesla' in info['architecture'].lower()
except TypeError:
# There might be than one GPUs, but for now we don't care
# as we're not really exploiting this info yet...
pass
def test_get_gpu_info(self):
info = get_gpu_info()
try:
assert info['architecture'].lower() in ['tesla', 'geforce', 'unspecified']
except KeyError:
# There might be than one GPUs, but for now we don't care
# as we're not really exploiting this info yet...
pass