"""Tests for general GPU support"""

import subprocess

import unittest

from common import gpu_test

class TestNvidia(unittest.TestCase):

@gpu_test

def test_system_management_interface(self):

smi = subprocess.Popen(['nvidia-smi'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

smi.communicate()

self.assertEqual(0, smi.returncode)