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