def test_NoChangeBMI(self):
"""
Expect BMI to remain after attempting to set it to 'other' (not
recognised in enumeration)
"""
expBMI = "Overweight"
rec = Record(None, "M")
"""Default values proven in other test"""
rec.setBMI(expBMI)
rec.setBMI("other")
self.assertEqual(expBMI, rec.getBMI())
def test_DefaultAttributes(self):
"""
Expect default atttibute values (age 0, sales 0, BMI Normal, income 0)
"""
expAge = 0
expSales = 0
expBMI = "Normal"
expIncome = 0
rec = Record(None, "M")
self.assertEqual(expAge, rec.getAge())
self.assertEqual(expSales, rec.getSales())
self.assertEqual(expBMI, rec.getBMI())
self.assertEqual(expIncome, rec.getIncome())
