def test_class_init():
''' Tests class Init for various success and failure cases'''
# Success Cases
assert Resource("Intel Core i9-9900K", "Intel", 10, 0)
assert Resource("Nvidia", "Nvidia", 0, 0)
assert Resource("Nvidia GTX", "Nvidia", 1, 0)
assert Resource("Intel Core i9-9900K", "Intel", 10, 10)
# Failure Cases
with pytest.raises(ValueError):
Resource("", "", 10, 0)
with pytest.raises(ValueError):
Resource("Nvidia", " ", 10, 0)
with pytest.raises(ValueError):
Resource("Nvidia GTX", "", 10, 0)
with pytest.raises(ValueError):
Resource("", "Nvidia", 10, 0)
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 15)
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", -10, 0)
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, -1)
with pytest.raises(TypeError):
Resource(12, "Intel", 10, 15)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", 14+3j, -10, 0)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", '10', 1)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10, -1+3j)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", Decimal('10'), 5)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10.5, 5)
def test_readonly_manufacturer():
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 0)
with pytest.raises(AttributeError):
r1.manufacturer = "NVIDIA"
def test_claim():
''' Tests the claim method of Resource Class'''
### Check Independent Updates
# Success Cases
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 2)
r1.claim(5)
assert r1.allocated == 7
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 6)
r1.claim(4)
assert r1.allocated == 10
# Failure Cases
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 0).claim(-1)
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 0).claim(11)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10, 0).claim(1.58)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10, 0).claim('11')
### Check Chained Updates (since this method mutates self)
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 2)
r1.claim(1)
assert r1.allocated == 3
r1.claim(2)
assert r1.allocated == 5
with pytest.raises(ValueError):
r1.claim(6)
def test_purchased():
''' Tests the purchase method of Resource Class'''
### Check Independent Updates
# Success Cases
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 7)
r1.purchased(5)
assert r1.total == 15
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 6)
r1.purchased(0)
assert r1.total == 10
# Failure Cases
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 6).purchased(-1)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10, 6).purchased(1.58)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10, 6).purchased('11')
### Check Chained Updates (since this method mutates self)
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 6)
r1.purchased(5)
assert r1.total == 15
r1.purchased(2)
assert r1.total == 17
def test_readonly_name():
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 0)
with pytest.raises(AttributeError):
r1.name = "NVIDIA GTX=1080"
def test_freeup():
''' Tests the freeup method of Resource Class'''
### Check Independent Updates
# Success Cases
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 7)
r1.freeup(5)
assert r1.allocated == 2
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 6)
r1.freeup(6)
assert r1.allocated == 0
# Failure Cases
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 7).freeup(-1)
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 7).freeup(8)
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 7).freeup(10)
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 7).freeup(12)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10, 7).freeup(1.58)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10, 7).freeup('11')
### Check Chained Updates (since this method mutates self)
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 7)
r1.freeup(5)
assert r1.allocated == 2
r1.freeup(1)
assert r1.allocated == 1
with pytest.raises(ValueError):
r1.freeup(5)
def test_kill():
''' Tests the kill method of Resource Class'''
### Check Independent Updates
# Success Cases
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 7)
r1.kill(5)
assert r1.total == 5
assert r1.allocated == 2
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 6)
r1.kill(2)
assert r1.total == 8
assert r1.allocated == 4
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 6)
r1.kill(8)
assert r1.total == 2
assert r1.allocated == 0
# Failure Cases
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 6).kill(-1)
with pytest.raises(ValueError):
Resource("Intel Core i9-9900K", "Intel", 10, 6).kill(11)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10, 6).kill(1.58)
with pytest.raises(TypeError):
Resource("Intel Core i9-9900K", "Intel", 10, 6).kill('11')
### Check Chained Updates (since this method mutates self)
r1 = Resource("Intel Core i9-9900K", "Intel", 10, 7)
r1.kill(5)
assert r1.total == 5
assert r1.allocated == 2
r1.kill(1)
assert r1.total == 4
assert r1.allocated == 1
with pytest.raises(ValueError):
r1.kill(5)
assert r1.allocated == 0
def test_create_invalid_allocated_value(total, allocated):
with pytest.raises(ValueError):
Resource('name', 'manu', total, allocated)
def test_create_invalid_total_value():
with pytest.raises(ValueError):
Resource('name', 'manu', -10, 0)
def test_create_invalid_allocated_type():
with pytest.raises(TypeError):
Resource('name', 'manu', 10, 2.5)
def test_create_invalid_total_type():
with pytest.raises(TypeError):
Resource('Parrot', 'Pirates A-Hoy', 10.5, 5)
def resource(resource_values):
return Resource(**resource_values)