Split pycuda test into its own file

rosbo · rosbo · commit 1423d613c481 · 2020-04-24T00:20:48.000Z
diff --git a/tests/test_nvidia.py b/tests/test_nvidia.py
@@ -1,8 +1,6 @@
 """Tests for general GPU support"""
 
-import os
 import subprocess
-import sys
 import unittest
 
 from common import gpu_test
@@ -14,11 +12,3 @@ def test_system_management_interface(self):
         smi = subprocess.Popen(['nvidia-smi'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
         smi.communicate()
         self.assertEqual(0, smi.returncode)
-
-    @gpu_test
-    def test_pycuda(self):   
-        import pycuda.driver    
-        pycuda.driver.init()
-        gpu_name = pycuda.driver.Device(0).name() 
-        self.assertNotEqual(0, len(gpu_name))
-
diff --git a/tests/test_pycuda.py b/tests/test_pycuda.py
@@ -0,0 +1,14 @@
+"""Tests for general GPU support"""
+
+import unittest
+
+from common import gpu_test
+
+
+class TestPycuda(unittest.TestCase):
+    @gpu_test
+    def test_pycuda(self):   
+        import pycuda.driver    
+        pycuda.driver.init()
+        gpu_name = pycuda.driver.Device(0).name() 
+        self.assertNotEqual(0, len(gpu_name))
