From fbe9240cf3bfc0b640713deb47db2be634ce40ab Mon Sep 17 00:00:00 2001 From: Axel Kohlmeyer Date: Tue, 11 May 2021 15:17:43 -0400 Subject: [PATCH] add python unit test for GPU capabilities --- unittest/python/python-capabilities.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/unittest/python/python-capabilities.py b/unittest/python/python-capabilities.py index 372eecc869..2c8aa8cdc3 100644 --- a/unittest/python/python-capabilities.py +++ b/unittest/python/python-capabilities.py @@ -158,5 +158,14 @@ class PythonCapabilities(unittest.TestCase): if self.cmake_cache['GPU_PREC'].lower() == 'single': 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__": unittest.main()