def test_set_virt_pxe_boot(test_input, expected_exception):
# Arrange
test_api = CobblerAPI()
test_manager = CollectionManager(test_api)
test_system = System(test_manager)
# Act
with expected_exception:
result = utils.set_virt_pxe_boot(test_system, test_input)
# Assert
assert test_system.virt_pxe_boot == 0 or test_system.virt_pxe_boot == 1
def set_virt_pxe_boot(self, num):
return utils.set_virt_pxe_boot(self, num)
