def test_remove_subordinate_id() -> None: e1 = Employee(1, "Emma Ployee", "Worker", 10000, 50) e2 = Employee(2, "Sue Perior", "Manager", 20000, 30) e1.become_subordinate(e2) assert e2.get_direct_subordinates()[0].eid == 1 e2.remove_subordinate_id(1) assert e2.get_direct_subordinates() == [] assert e1.get_superior() is e2
def test_remove_sub() -> None: e1 = Employee(5, "Emma Ployee", "Worker", 10000, 50) e2 = Employee(7, "Sue Perior", "Manager", 20000, 30) e3 = Employee(8, "Cowor Ker", "Worker", 10000, 50) e3.become_subordinate(e2) e1.become_subordinate(e2) assert len(e2.get_direct_subordinates()) == 2 e2.remove_subordinate_id(5) assert e1.get_superior() == e2 assert len(e2.get_direct_subordinates()) == 1 e1.become_subordinate(e3) assert len(e2.get_all_subordinates()) == 2 e2.remove_subordinate_id(5) assert len(e2.get_direct_subordinates()) == 1 assert e1.get_superior() == e3 e1.become_subordinate(e3) assert len(e2.get_all_subordinates()) == 2 e2.remove_subordinate_id(8) assert len(e2.get_all_subordinates()) == 0 assert e1.get_superior() == e3