Merge pull request #786 from Kaggle/split-pycuda

Split pycuda test into its own test file

Author: rosbo

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))
-

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))