def test_invalid_slot(self): slot_name = '01:00' pci_id = '1002:67df' try: pdev = PciDev(slot_name, pci_id, '') self.assertFalse(True) except: pdev = None
def test_mock(self): slot_name = '1111:11:11.1' pci_id = '1111:1111' pdev = PciDev(slot_name, pci_id, 'mock pci') self.assertIsNotNone(pdev) self.assertEqual(pdev.vendor_name(), 'Other') self.assertIsNotNone(pdev) gpu_dev = GpuDev(pdev) self.assertIsNotNone(gpu_dev)
def test_mock(self): slot_name = '1111:11:11.1' pci_id = '1111:1111' try: pdev = PciDev(slot_name, pci_id, 'mock pci') except: pdev = None self.assertIsNotNone(pdev) self.assertEqual(pdev.vendor_name(), 'Other')
def test_valid_slot(self): slot_name = '0000:01:00.0' pci_id = '1002:67df' try: pdev = PciDev(slot_name, pci_id, '') except: pdev = None self.assertIsNotNone(pdev) self.assertEqual(pdev.domain, '0000') self.assertEqual(pdev.bus_id, '01') self.assertEqual(pdev.slot_id, '00') self.assertEqual(pdev.function, '0')
def test_nak_gpu(self): slot_name = '1111:11:11.1' pci_id = '1111:1111' pdev = PciDev(slot_name, pci_id, 'mock pci') self.assertIsNotNone(pdev) self.assertEqual(pdev.vendor_name(), 'Other') gpu_dev = GpuNak(pdev) self.assertIsNotNone(gpu_dev) gpu_ctl = GpuCtl(gpu_devices=[gpu_dev], verbose=True) rv = gpu_ctl.set_interval(wait_period=10) self.assertTrue(rv) gpu_ctl.start() time.sleep(5) gpu_ctl.stop()
def test_interval(self): slot_name = '1111:11:11.1' pci_id = '1111:1111' pdev = PciDev(slot_name, pci_id, 'mock pci') self.assertIsNotNone(pdev) self.assertEqual(pdev.vendor_name(), 'Other') gpu_dev = GpuOk(pdev) self.assertIsNotNone(gpu_dev) gpu_ctl = GpuCtl(gpu_devices=[gpu_dev]) rv = gpu_ctl.set_interval(intvl=1, wait_period=3) self.assertTrue(rv) rv = gpu_ctl.set_interval(intvl=2, wait_period=20) self.assertTrue(rv) rv = gpu_ctl.set_interval(intvl=2, wait_period=1) self.assertFalse(rv)
def test_over_temp(self): slot_name = '1111:11:11.1' pci_id = '1111:1111' pdev = PciDev(slot_name, pci_id, 'mock pci') self.assertIsNotNone(pdev) self.assertEqual(pdev.vendor_name(), 'Other') gpu_dev = GpuNak(pdev) self.assertIsNotNone(gpu_dev) gpu_ctl = GpuCtl(gpu_devices=[gpu_dev], fan=10, temp=20, tas='./tests/ok.sh') self.assertNotEqual(gpu_ctl, None) rv = gpu_ctl.set_interval(wait_period=10) self.assertTrue(rv) gpu_ctl.start() time.sleep(5) gpu_ctl.stop()