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