def test_company_total_employee_salary(self):
company5 = Company("Apple5")
emp13 = Employee("Michael", title = "Software Enigineer", salary = 5000, rank = 3)
emp14 = Employee("Nick", title = "Software Enigineer", salary = 7000, rank = 2)
company5.addEmployee(emp13)
company5.addEmployee(emp14)
self.assertTrue(company5.getTotalSalaryofEmployee() == 12000)
def test_company_employee_count2(self):
company2 = Company("Apple2")
emp10 = Employee("Jack", title = "Secretary", salary = 3000, rank = 3)
emp11 = Employee("Kevin", title = "CTO", salary = 53000, rank = 2)
company2.addEmployee(emp10)
company2.addEmployee(emp11)
self.assertTrue(company2.getTotalNumOfEmployee() is 2)
def test_employee_job_hopping(self):
company3 = Company("Apple3")
company4 = Company("Apple4")
emp12 = Employee("Larry", title = "Software Enigineer", salary = 4000, rank = 3)
company3.addEmployee(emp12)
company3.removeEmployee(emp12)
company4.addEmployee(emp12)
self.assertTrue(company4.getTotalNumOfEmployee() is 1 and company3.getTotalNumOfEmployee() is 0)
def test_company_employee_count(self):
company1 = Company("Apple")
emp9 = Employee("Ian", title = "Program Manager", salary = 7000, rank = 4)
company1.addEmployee(emp9)
self.assertTrue(company1.getTotalNumOfEmployee() is 1)
