def test_line_obtain_subs() -> None:
o = Organization()
e1 = Employee(1, '', '', 1, 1)
o.add_employee(e1)
e2 = Employee(2, '', '', 1, 1)
o.add_employee(e2, 1)
e3 = Employee(3, '', '', 1, 1)
o.add_employee(e3, 1)
e4 = Employee(4, '', '', 1, 1)
o.add_employee(e4, 3)
e5 = Employee(5, '', '', 1, 1)
o.add_employee(e5, 4)
assert e2 in e1.get_direct_subordinates()
assert e3 in e1.get_direct_subordinates()
assert not e5 in e1.get_direct_subordinates()
assert e5 in e1.get_all_subordinates()
assert len(e2.get_all_subordinates()) == 0
e2.obtain_subordinates([3, 4])
assert not e3 in e1.get_direct_subordinates()
assert e2 in e1.get_direct_subordinates()
assert e3 in e2.get_direct_subordinates()
assert e4 in e2.get_direct_subordinates()
assert e5 in e1.get_direct_subordinates()
assert e3 in e2.get_direct_subordinates()
assert e4 in e2.get_direct_subordinates()
def test_obtain_employees() -> None:
o = Organization()
e1 = Leader(1, 'Holly', '', 1, 1, 'Department')
o.add_employee(e1)
e2 = Employee(2, 'Ivan', '', 1, 1)
o.add_employee(e2, 1)
e3 = Employee(3, 'Kevin', '', 1, 1)
o.add_employee(e3, 2)
e4 = Employee(4, 'Joe', 'Mama', 1, 100)
o.add_employee(e4, 2)
e5 = Employee(5, 'Linda', '', 1, 1)
o.add_employee(e5, 4)
assert e1.get_direct_subordinates() == [e2]
assert e2.get_superior() == e1
assert e2.get_direct_subordinates() == [e3, e4]
assert e3.get_superior() == e2
assert e4.get_superior() == e2
assert e3.get_direct_subordinates() == []
assert e4.get_direct_subordinates() == [e5]
assert e5.get_superior() == e4
e4.obtain_subordinates([2])
assert o.get_head() == e1
assert e1.get_direct_subordinates() == [e3, e4]
assert e3.get_superior() == e1
assert e4.get_superior() == e1
assert e3.get_direct_subordinates() == []
assert e4.get_direct_subordinates() == [e2, e5]
assert e2.get_superior() == e4
assert e5.get_superior() == e4
def test_get_average_salary_simple() -> None:
o = Organization()
assert o.get_average_salary() == 0.0
e1 = Employee(1, "Emma Ployee", "Worker", 10000, 50)
e2 = Employee(2, "Sue Perior", "Manager", 20000, 30)
o.add_employee(e2)
o.add_employee(e1, 2)
assert o.get_average_salary() == 15000.0
def test_short_create() -> None:
o = Organization()
lin = Leader(1, 'Linda', 'Sewed', 10, 10, 'MyNew')
o.add_employee(lin)
assert o.get_head() == lin
t = create_department_salary_tree(o)
assert t.department_name == 'MyNew'
assert size(t) == 1
def test_avg_sal() -> None:
org = Organization()
assert org.get_average_salary() == 0.0
e = Employee(1, 'Boss', 'CEO', 50000, 50)
org.add_employee(e)
e1 = Employee(2, 'Sarah', 'Keyboardist', 100000, 100)
org.add_employee(e1, 1)
assert org.get_average_salary() == 75000
assert org.get_average_salary('Keyboardist') == 100000
def test_fire_under_tie_simple() -> None:
e1 = Employee(1, '', '', 10, 10)
e2 = Employee(2, '', '', 10, 20)
e3 = Employee(3, '', '', 10, 20)
o = Organization()
o.add_employee(e1)
o.add_employee(e3, 1)
o.add_employee(e2, 1)
o.fire_under_rating(20)
assert o.get_head() == e2
assert e2.get_direct_subordinates() == [e3]
assert e3.get_superior() == e2
def test_department_salary_tree_simple() -> None:
e1 = Leader(1, "Sarah", "CEO", 300000, 89, "Some Corp.")
e2 = Leader(2, "Sandra", "Secretary", 20000, 89, "Finance")
e3 = Employee(3, "Sofia", "Manager", 300000, 89)
e4 = Leader(4, "Senya", "HR Head", 5000, 89, "Human Resources")
e5 = Employee(5, "Sylvia", "Secretary", 300000, 99)
e6 = Employee(6, "Selena", "Grunt", 5000, 99)
e7 = Employee(7, "Sophie", "Grunt", 5000, 99)
o = Organization()
o.add_employee(e1)
o.add_employee(e2, 1)
o.add_employee(e3, 1)
o.add_employee(e4, 1)
o.add_employee(e5, 1)
o.add_employee(e6, 4)
o.add_employee(e7, 4)
tree = create_department_salary_tree(o)
assert int(tree.salary) == 300000
assert tree.department_name == "Some Corp."
assert len(tree.subdepartments) == 2
assert int(tree.subdepartments[0].salary) == 20000
assert tree.subdepartments[0].department_name == "Finance"
assert len(tree.subdepartments[0].subdepartments) == 0
assert int(tree.subdepartments[1].salary) == 5000
assert tree.subdepartments[1].department_name == "Human Resources"
assert len(tree.subdepartments[1].subdepartments) == 0
def test_fire_head() -> None:
o = Organization()
e1 = Employee(1, "e1", "a", 1, 1)
e2 = Employee(2, "e2", "a", 1, 2)
e3 = Employee(3, "e3", "a", 1, 3)
e4 = Employee(4, "e4", "a", 1, 4)
e5 = Employee(5, "e5", "a", 1, 5)
e6 = Employee(6, "e6", "a", 1, 6)
e2.become_subordinate(e1)
e3.become_subordinate(e1)
e4.become_subordinate(e3)
e5.become_subordinate(e3)
e6.become_subordinate(e5)
o.add_employee(e1)
assert o.get_head() == e1
o.fire_employee(1)
assert o.get_head() == e3
o.fire_employee(3)
assert o.get_head() == e5
o.fire_employee(4)
assert o.get_head() == e5
o.fire_employee(5)
assert o.get_head() == e6
def test_fire_under_rating_advanced_2() -> None:
e1 = Leader(1, "Sarah", "CEO", 500000, 10, "Some Corp.")
e2 = Employee(2, "Sandra", "Secretary", 20000, 20)
e3 = Employee(3, "Sofia", "Manager", 25000, 30)
e4 = Employee(4, "Senya", "Grunt", 5000, 40)
e5 = Employee(5, "Sylvia", "Grunt", 5000, 50)
e6 = Employee(6, 'Savanahh', 'Stayin', 5000, 39)
o = Organization()
o.add_employee(e1)
o.add_employee(e2, 1)
o.add_employee(e3, 1)
o.add_employee(e4, 2)
o.add_employee(e6, 3)
o.add_employee(e5, 6)
o.fire_under_rating(40)
assert o.get_head() == e4
assert e4.get_direct_subordinates() == [e5]
assert e5.get_superior() == e4
def test_fire_lowest_employee_new_head() -> None:
e1 = Leader(1, "Sarah", "CEO", 500000, 1, "Some Corp.")
e2 = Employee(2, "Sandra", "Secretary", 20000, 99)
e3 = Employee(3, "Sofia", "Manager", 25000, 89)
e4 = Employee(4, "Senya", "Grunt", 5000, 89)
e5 = Employee(5, "Sylvia", "Grunt", 5000, 89)
e6 = Employee(6, "Savannah", "Wedding Officiate", 5000, 50)
o = Organization()
o.add_employee(e1)
o.add_employee(e2, 1)
o.add_employee(e3, 1)
o.add_employee(e4, 3)
o.add_employee(e5, 3)
o.add_employee(e6, 2)
o.fire_lowest_rated_employee()
assert o.get_head().eid == 2
assert o.get_head().get_direct_subordinates() == [e3, e6]
assert e3.get_superior() == e2
def test_get_employee_advanced() -> None:
e1 = Employee(1, "Emma Ployee", "Worker", 10000, 50)
e2 = Employee(2, "Sue Perior", "Manager", 50000, 30)
e3 = Employee(3, "Robocop", "Worker", 20000, 30)
e4 = Leader(4, "Sarah", "CEO", 500000, 30, "Some Corp.")
e5 = Employee(5, "Sofia", "Worker", 10000, 30)
e6 = Employee(6, "Terry", "Worker", 5000, 30)
e7 = Employee(7, "Odysseus", "Worker", 22000, 30)
o = Organization()
o.add_employee(e4)
o.add_employee(e7, 4)
o.add_employee(e6, 4)
o.add_employee(e2, 4)
o.add_employee(e1, 2)
o.add_employee(e3, 2)
o.add_employee(e5, 2)
assert o.get_employee(1) is e1
assert o.get_employee(4) is e4
assert o.get_employee(12) is None
def test_add_employee_simple() -> None:
o = Organization()
e1 = Employee(1, "Emma Ployee", "Worker", 10000, 50)
e2 = Employee(2, "Sue Perior", "Manager", 20000, 30)
o.add_employee(e2)
assert o.get_head() is e2
o.add_employee(e1, 2)
assert o.get_employee(1) is e1
assert e1.get_superior() is e2
e3 = Employee(3, "Bruce Wayne", "Philanthropist", 0, 100)
o.add_employee(e3)
assert o.get_head() == e3
def test_get_employees_with_position() -> None:
e1 = Employee(1, "Emma Ployee", "Worker", 10000, 50)
e2 = Employee(2, "Sue Perior", "Manager", 50000, 30)
e3 = Employee(3, "Robocop", "Worker", 20000, 30)
e4 = Leader(4, "Sarah", "CEO", 500000, 30, "Some Corp.")
e5 = Employee(5, "Sofia", "Worker", 10000, 30)
e6 = Employee(6, "Terry", "Worker", 5000, 30)
e7 = Employee(7, "Odysseus", "Worker", 22000, 30)
o = Organization()
o.add_employee(e4)
o.add_employee(e7, 4)
o.add_employee(e6, 4)
o.add_employee(e2, 4)
o.add_employee(e1, 2)
o.add_employee(e3, 2)
o.add_employee(e5, 3)
assert o.get_employees_with_position("Worker") == [e1, e3, e5, e6, e7]
assert o.get_employees_with_position("Manager") == [e2]
assert o.get_employees_with_position("CEO") == [e4]
def test_get_average_salary_advanced() -> None:
e1 = Employee(1, "Emma Ployee", "Worker", 50000, 50)
e2 = Employee(2, "Sue Perior", "Manager", 50000, 30)
e3 = Employee(3, "Robocop", "Worker", 50000, 30)
e4 = Leader(4, "Sarah", "CEO", 500000, 30, "Some Corp.")
e5 = Employee(5, "Sofia", "Worker", 50000, 30)
o = Organization()
o.add_employee(e4)
o.add_employee(e2, 4)
o.add_employee(e1, 2)
o.add_employee(e3, 2)
o.add_employee(e5, 2)
assert round(o.get_average_salary()) == 140000
def test_add_emp() -> None:
org = Organization()
e = Employee(1, 'Boss', 'CEO', 50000, 50)
assert org.get_employee(1) is None
org.add_employee(e)
assert org.get_employee(1).name == 'Boss'
e1 = Employee(2, 'Sarah', 'Keyboardist', 1000000, 100)
org.add_employee(e1, 1)
assert org.get_employee(2).name == 'Sarah'
assert e1.get_superior() == e
assert len(e.get_all_subordinates()) == 1
e2 = Employee(3, 'John', 'Doe', 10, 10)
org.add_employee(e2, 2)
assert org.get_employee(3).name == 'John'
assert e2.get_superior() == e1
def test_fire_under_rating_simple() -> None:
e1 = Leader(1, "Sarah", "CEO", 500000, 99, "Some Corp.")
e2 = Employee(2, "Sandra", "Secretary", 20000, 99)
e3 = Employee(3, "Sofia", "Manager", 25000, 89)
e4 = Employee(4, "Senya", "Grunt", 5000, 89)
e5 = Employee(5, "Sylvia", "Grunt", 5000, 89)
o = Organization()
o.add_employee(e1)
o.add_employee(e2, 1)
o.add_employee(e3, 1)
o.add_employee(e4, 3)
o.add_employee(e5, 3)
o.fire_under_rating(99)
assert o.get_head() == e1
assert e1.get_direct_subordinates() == [e2]
def test_fire_under_tie_advanced() -> None:
e1 = Employee(1, '', '', 10, 10)
e2 = Employee(2, '', '', 10, 10)
e3 = Employee(3, '', '', 10, 20)
e4 = Employee(4, '', '', 10, 10)
e5 = Employee(5, '', '', 10, 20)
o = Organization()
o.add_employee(e1)
o.add_employee(e5, 1)
o.add_employee(e2, 5)
o.add_employee(e4, 1)
o.add_employee(e3, 4)
o.fire_under_rating(20)
assert o.get_head() == e5
assert e5.get_direct_subordinates() == [e3]
assert e3.get_superior() == e5
def test_fire_lowest_rated_employee_tie() -> None:
e1 = Leader(1, "Sarah", "CEO", 500000, 99, "Some Corp.")
e2 = Employee(2, "Sandra", "Secretary", 20000, 99)
e3 = Employee(3, "Sofia", "Manager", 25000, 89)
e4 = Employee(4, "Senya", "Grunt", 5000, 89)
e5 = Employee(5, "Sylvia", "Grunt", 5000, 89)
o = Organization()
o.add_employee(e1)
o.add_employee(e2, 1)
o.add_employee(e3, 1)
o.add_employee(e4, 3)
o.add_employee(e5, 3)
o.fire_lowest_rated_employee()
assert e1.get_direct_subordinates() == [e2, e4, e5]
assert e4.get_superior() == e1
assert e5.get_superior() == e1
def test_medium_create() -> None:
o = Organization()
l1 = create_lead(1)
l2 = create_lead(2)
l3 = create_lead(3)
o.add_employee(l1)
o.add_employee(l2, 1)
o.add_employee(l3, 1)
e4 = create_emp(4)
e5 = create_emp(5)
e6 = create_emp(6)
e7 = create_emp(7)
o.add_employee(e4, 2)
o.add_employee(e5, 2)
o.add_employee(e6, 3)
o.add_employee(e7, 3)
dept_hier = create_department_salary_tree(o)
assert dept_hier.department_name == 'Dept1'
subdepts = dept_hier.subdepartments
assert len(subdepts) == 2
salaries = [subdepts[0].salary, subdepts[1].salary]
assert 2900.0 / 3 in salaries
assert 4300.0 / 3 in salaries
assert size(dept_hier) == 3
def test_fire_employee_new_head_2() -> None:
e1 = Leader(1, "Sarah", "CEO", 500000, 30, "Some Corp.")
e2 = Employee(2, "Sandra", "Secretary", 20000, 40)
e3 = Employee(3, "Sofia", "Manager", 25000, 30)
e4 = Employee(4, "Senya", "Grunt", 5000, 30)
e5 = Employee(5, "Sylvia", "Grunt", 5000, 40)
o = Organization()
o.add_employee(e1)
o.add_employee(e2, 1)
o.add_employee(e3, 1)
o.add_employee(e4, 3)
o.add_employee(e5, 3)
o.fire_employee(1)
head = o.get_head()
assert head.eid == 2
assert head.get_direct_subordinates() == [e3]
assert e3.get_superior() == head
def test_promote_employee_simple() -> None:
e1 = Leader(1, "Sarah", "CEO", 500000, 89, "Some Corp.")
e2 = Employee(2, "Sandra", "Secretary", 20000, 89)
e3 = Employee(3, "Sofia", "Manager", 25000, 89)
e4 = Employee(4, "Senya", "Grunt", 5000, 89)
e5 = Employee(5, "Sylvia", "Grunt", 5000, 99)
o = Organization()
o.add_employee(e1)
o.add_employee(e2, 1)
o.add_employee(e3, 1)
o.add_employee(e4, 3)
o.add_employee(e5, 3)
o.promote_employee(5)
assert o.get_head().eid == 5
assert len(o.get_head().get_direct_subordinates()) == 2
assert o.get_head().get_direct_subordinates()[0].eid == 1
assert o.get_head().get_direct_subordinates()[1].eid == 2
def test_empty_create() -> None:
o = Organization()
assert create_department_salary_tree(o) is None
lin = Employee(1, 'Linda', 'Tailor', 10, 10)
o.add_employee(lin)
assert create_department_salary_tree(o) is None
def add_emps(org: Organization, emps: [Employee]) -> None:
for emp in emps:
org.add_employee(emp)
class OrganizationSimulator:
"""OrganizationSimulator: a class that stores information about the current
state of a simulated organization.
=== Public Attributes ===
current_employee:
The Employee currently beieng displayed in the simulation.
current_subordinates:
The list of subordinates currently being displayed in the simulation.
displaying_direct:
Whether the current_subordinates being displayed are only the direct
subordinates of current_employee or not.
current_organization:
The Organization currently being displayed in the simulation.
"""
current_employee: Optional[Employee]
current_subordinates: List[Employee]
displaying_direct: bool
current_organization: Organization
def __init__(self) -> None:
"""Initialize this OrganizationSimulator with an empty Organization.
"""
self.current_organization = Organization()
self.current_employee = None
self.current_subordinates = []
self.displaying_direct = True
# TODO: === TASK 1 ===
# Go through the below methods to find methods that you'll need to implement
# in organization_hierarchy.py.
def display_employee(self, eid: int) -> None:
"""Update self.current_employee to be the employee with the ID <eid>.
If no such employee exist, self.current_employee should be None.
"""
self.current_employee = self.current_organization.get_employee(eid)
def get_current_employee_details(self) -> Tuple[str, float, int, str, str]:
"""Return the self.current_employee's name, salary, position, and
department.
"""
if self.current_employee:
return (self.current_employee.name, self.current_employee.salary,
self.current_employee.rating,
self.current_employee.position,
self.get_current_employee_department())
return "", 0.0, 0, "", ""
def get_current_superior(self) -> Optional[Employee]:
"""Return the superior of self.current_employee.
Pre-condition: self.current_employee has a superior (i.e. they are not
the head of the organization.)
"""
if self.current_employee:
return self.current_employee.get_superior()
return None
def create_employee(self, name: str, eid: int, salary: float, rating: int,
position: str, superior_id: int) -> None:
"""Create the Employee with the name <name>, ID <eid>, salary <salary>,
position <position>, rating <rating>, and set their superior to be the
employee with ID <superior_id>.
If <superior_id> == 0, this employee is the new head of the
organization.
If <eid> is already in use or < 0, then pick the next free ID number as
the ID.
Pre-condition: <superior_id> == 0 or is an ID that appears in the
organization.
"""
# Find the next free id if eid is already taken or is < 1
if eid < 1 or self.current_organization.get_employee(eid) is not None:
eid = self.current_organization.get_next_free_id()
# Create the employee
new_employee = Employee(eid, name, position, salary, rating)
# Add the employee to the organization
if superior_id <= 0:
self.current_organization.add_employee(new_employee)
else:
self.current_organization.add_employee(new_employee, superior_id)
self.display_employee(eid)
def find_employees_with_position(self, position: str) -> List[Employee]:
"""Return a list of employees in self.current_organization with the
position named <position> in order of increasing eids.
"""
return self.current_organization.get_employees_with_position(position)
def get_average_salary(self) -> float:
"""Return the average salary of all employees in
self.current_organization.
If there are no employees in self.current_organization, return 0.0
"""
return self.current_organization.get_average_salary()
def get_average_salary_for_position(self, position: str) -> float:
"""Return self.current_organization's average salary for all employees
with the position <position>. Return 0.0 if no such employees exist.
"""
return self.current_organization.get_average_salary(position)
def display_direct_subordinates(self) -> None:
"""Update the list of subordinates to display self.current_employee's
direct subordinates.
Subordinates must be in order of increasing eids.
"""
self.displaying_direct = True
if self.current_employee:
self.current_subordinates = \
self.current_employee.get_direct_subordinates()
def display_all_subordinates(self) -> None:
"""Update the list of subordinates to display all of
self.current_employee's subordinates.
Subordinates must be in order of increasing eids.
"""
self.displaying_direct = False
if self.current_employee:
self.current_subordinates = \
self.current_employee.get_all_subordinates()
def get_higher_paid_employees(self) -> List[Employee]:
"""Return a list of all employees in the Organization that are paid more
than self.current_employee.
Employees must be returned with IDs in increasing order.
"""
return self.current_employee.get_higher_paid_employees()
def get_organization_head(self) -> Employee:
"""Return the head of the organization.
The head of the organization is defined as the employee that does not
have any superiors.
"""
return self.current_employee.get_organization_head()
def get_common_superior(self, eid: int) -> Employee:
"""Return the closest common superior in the organization between
self.current_employee and the employee with ID <eid>.
Precondition: <eid> exists in the organization.
"""
return self.current_employee.get_closest_common_superior(eid)
# TODO: === TASK 2 ===
# Go through the below methods to find methods that you'll need to implement
# in organization_hierarchy.py.
def is_leader(self) -> bool:
"""Return True if self.current_employee is a Leader.
"""
return isinstance(self.current_employee, Leader)
def get_current_employee_department(self) -> str:
"""Return the department that self.current_employee belongs to.
If self.current_employee is not part of any department, return "".
"""
if self.current_employee:
return self.current_employee.get_department_name()
return ""
def get_position_in_hierarchy(self) -> str:
"""Return the full position of self.current_employee.
The full position takes the form:
<self.current_employee's position>
Followed by a comma separating any departments they're a part of (from
their immediate department, to the department that one belongs to, and
so on.)
"""
return self.current_employee.get_position_in_hierarchy()
def find_department_employees(self) -> List[Employee]:
"""Return a list of employees in self.current_employee's department.
Pre-condition: self.current_employee is a Leader.
"""
return self.current_employee.get_department_employees()
# TODO: === TASK 3 ===
# Go through the below methods to find methods that you'll need to implement
# in organization_hierarchy.py.
def take_over_department(self) -> None:
"""Makes self.current_employee the leader of their current department,
becoming the superior of the current department leader.
self.current_employee keeps all of their subordinates, in addition
to gaining the leader as a subordinate.
If self.current_employee is already a leader or does not belong to a
department, nothing happens.
"""
# Keep the employee's ID number so we can update self.current_employee
# to the leader version afterwards.
eid = self.current_employee.eid
organization_head = self.current_employee.change_department_leader()
self.current_organization.set_head(organization_head)
# Update the current employee being displayed
self.current_employee = self.current_organization.get_employee(eid)
def become_leader(self, department_name: str) -> None:
"""Make self.current_employee the leader of a new department with the
name <department_name>.
If self.current_employee is already a leader, changes the name of their
department to department_name.
"""
# Keep the employee's ID number so we can update self.current_employee
# to the leader version afterwards.
eid = self.current_employee.eid
leader = self.current_employee.become_leader(department_name)
self.current_organization.set_head(leader.get_organization_head())
# Update the current employee being displayed
self.current_employee = self.current_organization.get_employee(eid)
def become_employee(self) -> None:
"""Make self.current_employee an employee instead of a leader.
Pre-condition: self.current_employee is a Leader.
"""
# Keep the employee's ID number so we can update self.current_employee
# to the employee version afterwards.
eid = self.current_employee.eid
employee = self.current_employee.become_employee()
self.current_organization.set_head(employee.get_organization_head())
# Update the current employee being displayed
self.current_employee = self.current_organization.get_employee(eid)
# TODO: === TASK 4 ===
# Go through the below methods to find methods that you'll need to implement
# in organization_hierarchy.py.
def obtain_subordinates(self, ids: List[int]) -> None:
"""Set the employees with IDs in ids as subordinates of
self.current_employee.
If those employees have subordinates, the superior of those subordinates
becomes the employee's original superior.
Pre-condition: self.current_employee's id is not in ids.
"""
organization_head = self.current_employee.obtain_subordinates(ids)
self.current_organization.set_head(organization_head)
def fire_employee(self, eid: int) -> None:
"""Fire the employee with ID eid from self.current_organization.
Pre-condition: there is an employee with the eid <eid> in
self.current_organization.
"""
previous_id = self.current_employee.eid
self.current_organization.fire_employee(eid)
if previous_id == eid:
self.current_employee = self.current_organization.get_head()
def fire_lowest_rated_employee(self) -> None:
"""Fire the lowest rated employee in self.current_organization.
If two employees have the same rating, the one with the lowest id
is fired.
"""
self.current_organization.fire_lowest_rated_employee()
# Update the display if the previous self.current_employee is no longer
# in the organization
eid = self.current_employee.eid
if not self.current_organization.get_employee(eid):
self.current_employee = self.current_organization.get_head()
def fire_under_rating(self, rating: int) -> None:
"""Fire all employees with a rating below rating.
Employees should be fired in order of increasing rating: the lowest
rated employees are to be removed first. Break ties in order of eid.
"""
self.current_organization.fire_under_rating(rating)
# Update the display if the previous self.current_employee is no longer
# in the organization
eid = self.current_employee.eid
if not self.current_organization.get_employee(eid):
self.current_employee = self.current_organization.get_head()
def promote_employee(self, eid: int) -> None:
"""Promote the employee with the eid <eid> in self.current_organization
until they have a superior with a higher rating than them or until they
are the head of the organization.
Precondition: There is an employee in self.current_organization with
eid <eid>.
"""
self.current_organization.promote_employee(eid)
self.current_employee = \
self.current_organization.get_employee(self.current_employee.eid)
# === TASK 5 ===
# TODO: Implement create_department_salary_tree in organization_hierarchy.py
def get_department_salary_tree(self) -> DepartmentSalaryTree:
"""Return the DepartmentSalaryTree that corresponds to
self.current_organization.
"""
return create_department_salary_tree(self.current_organization)
# === TASK 6 ===
# TODO: Implement create_organization_from_file in organization_hierarchy.py
def file_to_organization(self, filename: str) -> None:
"""Read the organization data in the file named filename, creating an
organization from it and setting it as self.current_organization.
Set the current employee to the head of the organization.
"""
f = open(filename)
self.current_organization = create_organization_from_file(f)
self.current_employee = self.current_organization.get_head()
def test_get_average_salary_position_filter() -> None:
e1 = Employee(1, "Emma Ployee", "Worker", 20000, 50)
e2 = Employee(2, "Sue Perior", "Manager", 50000, 30)
e3 = Employee(3, "Robocop", "Worker", 30000, 30)
e4 = Leader(4, "Sarah", "CEO", 500000, 30, "Some Corp.")
e5 = Employee(5, "Sofia", "Worker", 10000, 30)
e6 = Employee(6, "Terry", "Worker", 20000, 30)
e7 = Employee(7, "Odysseus", "Worker", 20000, 30)
e8 = Employee(8, "Sparrow", "Contract Worker", 60000, 30)
e9 = Employee(9, "Nick", "Contract Worker", 70000, 30)
e10 = Employee(10, "Theseus", "Contract Worker", 80000, 30)
o = Organization()
o.add_employee(e4)
o.add_employee(e2, 4)
o.add_employee(e1, 2)
o.add_employee(e3, 2)
o.add_employee(e5, 2)
o.add_employee(e6, 4)
o.add_employee(e7, 4)
o.add_employee(e8, 1)
o.add_employee(e9, 1)
o.add_employee(e10, 1)
assert round(o.get_average_salary("Worker")) == 20000
def test_get_next_free_id() -> None:
dans = Organization()
dans.add_employee(Employee(1, 'Dan', 'Iel', 1, 1))
dans.add_employee(Employee(2, 'Dan', 'Iel', 1, 1), 1)
dans.add_employee(Employee(3, 'Dan', 'Iel', 1, 1), 1)
dans.add_employee(Employee(4, 'Dan', 'Iel', 1, 1), 3)
dans.add_employee(Employee(6, 'Dan', 'Iel', 1, 1), 3)
dans.add_employee(Employee(100, 'Dan', 'Iel', 1, 1), 6)
dans.add_employee(Employee(5, 'Dan', 'Iel', 1, 1), 2)
dans.add_employee(Employee(8, 'Dan', 'Iel', 1, 1), 100)
dans.add_employee(Employee(13, 'Dan', 'Iel', 1, 1), 2)
dans.add_employee(Employee(27, 'Dan', 'Iel', 1, 1), 13)
assert dans.get_next_free_id() not in [1, 2, 3, 4, 5, 6, 8, 13, 27, 100]
def test_med_2_create() -> None:
leads = create_leaders(4)
emps = create_emps(5)
o = Organization()
e = leads[0]
o.add_employee(e)
o.add_employee(leads[1], 1)
o.add_employee(leads[2], 1)
o.add_employee(leads[3], 2)
o.add_employee(emps[0], 4)
o.add_employee(emps[1], 4)
o.add_employee(emps[2], 2)
o.add_employee(emps[3], 3)
o.add_employee(emps[4], 104)
d = create_department_salary_tree(o)
assert size(d) == 4
def test_lead_emp_emp_lead() -> None:
o = Organization()
leads = create_leaders(4)
emps = create_emps(4)
o.add_employee(leads[0])
o.add_employee(leads[1], 1)
o.add_employee(emps[0], 1)
o.add_employee(emps[1], 101)
o.add_employee(leads[2], 102)
o.add_employee(emps[2], 3)
o.add_employee(leads[3], 1)
tree = create_department_salary_tree(o)
assert size(tree) == 4
assert tree.department_name == 'Department1'
def test_get_employee_simple() -> None:
o = Organization()
e1 = Employee(1, "Emma Ployee", "Worker", 10000, 50)
o.add_employee(e1)
assert o.get_employee(1) is e1
assert o.get_employee(2) is None