def test_return_dict(self):
elec = ElectricAppliances(54, 'Test Elec', 200, 300, 'John Deere', 2)
elec_dict = elec.return_as_dictionary()
test_dict = {'product_code': 54,
'description': 'Test Elec',
'market_price': 200,
'rental_price': 300,
'brand': 'John Deere',
'voltage': 2}
self.assertEqual(elec_dict, test_dict)
def test_init(self):
""" Test init """
test_inventory = ElectricAppliances(123, "chair", 100, 50, "GE", 75)
self.assertEqual(test_inventory.brand, "GE")
self.assertEqual(test_inventory.voltage, 75)
def add_new_item():
"""add new item"""
item_code = input("Enter item code: ")
item_description = input("Enter item description: ")
item_rental_price = input("Enter item rental price: ")
# Get price from the market prices module
item_price = get_latest_price()
is_furniture = input("Is this item a piece of furniture? (Y/N): ")
if is_furniture.lower() == "y":
item_material = input("Enter item material: ")
item_size = input("Enter item size (S,M,L,XL): ")
new_item = Furniture(item_code, item_description, item_price,
item_rental_price, item_material, item_size)
else:
is_electric_appliance = input(
"Is this item an electric appliance? (Y/N): ")
if is_electric_appliance.lower() == "y":
item_brand = input("Enter item brand: ")
item_voltage = input("Enter item voltage: ")
new_item = ElectricAppliances(item_code, item_description,
item_price, item_rental_price,
item_brand, item_voltage)
else:
new_item = Inventory(item_code, item_description, item_price,
item_rental_price)
FULL_INVENTORY[item_code] = new_item.return_as_dictionary()
print("New inventory item added")
class inventory_test(TestCase):
"""unit test for inventory and its sub classes"""
def setUp(self):
"""set up instance of each sub class"""
self.furniture = Furniture(100, "test chair", 20, (1.5, "day"), "wood",
"Large")
self.appliance = ElectricAppliances(200, "test lamp", 30, (2.5, "day"),
"basic", "110V")
self.appliance_expected_output = {
"product_code": 200,
"description": "test lamp",
"market_price": 30,
"rental_price": (2.5, "day"),
"brand": "basic",
"voltage": "110V",
}
self.furniture_expected_output = {
"product_code": 100,
"description": "test chair",
"market_price": 20,
"rental_price": (1.5, "day"),
"material": "wood",
"size": "Large",
}
def test_electric_appliances(self):
self.assertDictEqual(self.appliance.return_as_dictionary(),
self.appliance_expected_output)
def test_furniture(self):
self.assertDictEqual(self.furniture.return_as_dictionary(),
self.furniture_expected_output)
def test_return_as_dictionary(self):
""" Test the return as dictionary method """
test_inventory = ElectricAppliances(123, "chair", 100, 50, "GE",
75).return_as_dictionary()
self.assertEqual(test_inventory['brand'], 'GE')
self.assertEqual(test_inventory['voltage'], 75)
def setUp(self):
"""set up instance of each sub class"""
self.furniture = Furniture(100, "test chair", 20, (1.5, "day"), "wood",
"Large")
self.appliance = ElectricAppliances(200, "test lamp", 30, (2.5, "day"),
"basic", "110V")
self.appliance_expected_output = {
"product_code": 200,
"description": "test lamp",
"market_price": 30,
"rental_price": (2.5, "day"),
"brand": "basic",
"voltage": "110V",
}
self.furniture_expected_output = {
"product_code": 100,
"description": "test chair",
"market_price": 20,
"rental_price": (1.5, "day"),
"material": "wood",
"size": "Large",
}
def test_elec_init(self):
elec = ElectricAppliances(54, 'Test Elec', 200, 300, 'John Deere', 2)
self.assertEqual(elec.brand, 'John Deere')
self.assertEqual(elec.voltage, 2)