add python unit test for GPU capabilities

2021-05-11 15:17:43 -04:00 · 2021-05-11 15:17:43 -04:00 · fbe9240cf3
parent 228711f59b
commit fbe9240cf3
1 changed files with 9 additions and 0 deletions
--- 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()