class ElectricApplianceTests(TestCase):
"""Tests for the ElectricAppliances class."""
def setUp(self):
self.appl = ElectricAppliances("A1", "Test", 2, 3, "Brand", 4)
def test_init(self):
"""Test object instantiation."""
self.assertEqual(self.appl.product_code, "A1")
self.assertEqual(self.appl.description, "Test")
self.assertEqual(self.appl.market_price, 2)
self.assertEqual(self.appl.rental_price, 3)
self.assertEqual(self.appl.brand, "Brand")
self.assertEqual(self.appl.voltage, 4)
def test_return_dict(self):
"""Test return_as_dict method."""
# expected dict to return
appl_dict = {
'product_code': "A1",
'description': "Test",
'market_price': 2,
'rental_price': 3,
'brand': "Brand",
'voltage': 4
}
self.assertEqual(appl_dict, self.appl.return_as_dict())
class TestElectricAppliances(TestCase):
""" tests modules in electric_appliances_class.py """
def setUp(self):
self.product_code = 199021
self.description = 'na'
self.market_price = 100
self.rental_price = 500
self.brand = 'ge'
self.voltage = 120
self.check_electric = ElectricAppliances(self.product_code,
self.description,
self.market_price,
self.rental_price, self.brand,
self.voltage)
def test_init(self):
""" check initialization of variables """
self.assertEqual(self.check_electric.product_code, self.product_code)
self.assertEqual(self.check_electric.description, self.description)
self.assertEqual(self.check_electric.market_price, self.market_price)
self.assertEqual(self.check_electric.rental_price, self.rental_price)
self.assertEqual(self.check_electric.brand, self.brand)
self.assertEqual(self.check_electric.voltage, self.voltage)
def test_return_as_dict(self):
""" check dictionary output """
dict_output = self.check_electric.return_as_dict()
self.assertEqual(dict_output['product_code'], self.product_code)
self.assertEqual(dict_output['description'], self.description)
self.assertEqual(dict_output['market_price'], self.market_price)
self.assertEqual(dict_output['rental_price'], self.rental_price)
self.assertEqual(dict_output['brand'], self.brand)
self.assertEqual(dict_output['voltage'], self.voltage)
