Add tests

Author: mdboom

cuda_core/cuda/core/system/_mig.pxi

@@ -27,7 +27,7 @@ cdef class MigInfo:
 
         For Ampere™ or newer fully supported devices.
         """
-        return nvml.device_is_mig_device_handle(self._device._handle)
+        return bool(nvml.device_is_mig_device_handle(self._device._handle))
 
     @property
     def mode(self) -> bool:

cuda_core/tests/system/test_system_device.py

@@ -729,3 +729,30 @@ def test_pstates():
             assert isinstance(utilization.percentage, int)
             assert isinstance(utilization.inc_threshold, int)
             assert isinstance(utilization.dec_threshold, int)
+
+
+@pytest.mark.skipif(helpers.IS_WSL or helpers.IS_WINDOWS, reason="MIG not supported on WSL or Windows")
+def test_mig():
+    for device in system.Device.get_all_devices():
+        with unsupported_before(device, DeviceArch.AMPERE):
+            mig = device.mig
+
+            assert isinstance(mig.is_mig_device, bool)
+            if mig.is_mig_device:
+                assert isinstance(mig.mode, bool)
+                assert isinstance(mig.pending_mode, bool)
+
+                device_count = mig.get_device_count()
+                assert isinstance(device_count, int)
+                assert device_count >= 0
+
+                for mig_device in mig.get_all_devices():
+                    assert isinstance(mig_device, system.Device)
+
+
+def test_uuid_with_prefix():
+    for device in system.Device.get_all_devices():
+        uuid_with_prefix = device.uuid_with_prefix
+        assert isinstance(uuid_with_prefix, str)
+        assert uuid_with_prefix.startswith(("GPU-", "MIG-"))
+        assert uuid_with_prefix[4:] == device.uuid