ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add python unit test for GPU capabilities

Browse Source
This commit is contained in:
Axel Kohlmeyer
2021-05-11 15:17:43 -04:00
parent 228711f59b
commit fbe9240cf3
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
unittest/python/python-capabilities.py
View File

@ -158,5 +158,14 @@ class PythonCapabilities(unittest.TestCase):
if self.cmake_cache['GPU_PREC'].lower() == 'single': if self.cmake_cache['GPU_PREC'].lower() == 'single':
self.assertIn('single',settings['GPU']['precision']) self.assertIn('single',settings['GPU']['precision'])
def test_gpu_device(self):
info = self.lmp.get_gpu_device_info()
if self.lmp.has_gpu_device:
self.assertTrue(info)
self.assertGreaterEqual(info.find("Device"),0)
else:
self.assertFalse(info)
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()