def servers(self):
return [
types.ServerConfig(
id="myid1",
name="Server1",
address="127.0.0.1:3123",
devices=[
types.Device(id="cpu0", name="MyCpu1", enabled=True),
types.Device(id="gpu1", name="GPU1", enabled=False),
],
)
]
def test_device_list_from_config(self, qtbot, form):
config = types.ServerConfig.default(devices=[
types.Device(enabled=True, id="cpu", name="CPU"),
types.Device(enabled=False, id="gpu:0", name="GPU 12"),
])
form.config = config
assert 2 == form.deviceList.count()
for idx in range(form.deviceList.count()):
widget_item = form.deviceList.item(idx)
device = config.devices[idx]
assert bool(widget_item.checkState()) is device.enabled
assert device.name in widget_item.text()
def servers(self):
return [
types.ServerConfig(id='myid1',
name='Server1',
type='local',
address='127.0.0.1',
port='3123',
devices=[
types.Device(id='cpu0',
name='MyCpu1',
enabled=True),
types.Device(id='gpu1',
name='GPU1',
enabled=False)
])
]
def test_device_list(self, qtbot, form):
qtbot.mouseClick(form.getDevicesBtn, Qt.LeftButton)
expected_devices = [types.Device(enabled=False, id=id_, name=name) for id_, name in self.DEVICE_LIST]
assert expected_devices == form.config.devices
test_item = form.deviceList.item(1)
click_pos = form.deviceList.visualItemRect(test_item).center()
qtbot.mouseClick(form.deviceList.viewport(), Qt.LeftButton, pos=click_pos)
assert form.config.devices[1].enabled
def test_device_list_merging(self, qtbot, form):
config = types.ServerConfig.default(devices=[
types.Device(enabled=True, id="gpu:1", name="GPU 12"),
])
form.config = config
qtbot.mouseClick(form.getDevicesBtn, Qt.LeftButton)
assert 3 == form.deviceList.count()
has_gpu1 = False
for idx in range(form.deviceList.count()):
widget_item = form.deviceList.item(idx)
if widget_item.id == "gpu:1":
assert widget_item.checkState()
has_gpu1 = True
assert has_gpu1
