def test_get_employees_paid_more_than_simple() -> None: e1 = Employee(1, "Emma Ployee", "Worker", 10000, 50) e2 = Employee(2, "Sue Perior", "Manager", 20000, 30) e3 = Leader(3, "Bigg Boss", "CEO", 50000, 60, "Some Corp.") e1.become_subordinate(e2) e2.become_subordinate(e3) more_than_10000 = e3.get_employees_paid_more_than(10000) assert len(more_than_10000) == 2 assert more_than_10000[0].name == 'Sue Perior' assert more_than_10000[1].name == 'Bigg Boss'
def test_get_employees_paid_more_than_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) e1.become_subordinate(e2) e3.become_subordinate(e2) e2.become_subordinate(e4) e5.become_subordinate(e2) e6.become_subordinate(e4) e7.become_subordinate(e4) more_than_20000 = e4.get_employees_paid_more_than(20000) assert more_than_20000 == [e2, e4, e7]