def test__set_captain_captain_to_captain(): user_picks = get_picks() captain = next(p for p in user_picks if p["is_captain"]) _set_captain(user_picks, 270, "is_captain", user_player_ids) new_captain = next(p for p in user_picks if p["is_captain"]) assert new_captain["element"] == captain["element"]
def test__set_captain_vice_to_vice(): user_picks = get_picks() vice_captain = next(p for p in user_picks if p["is_vice_captain"]) _set_captain(user_picks, 302, "is_vice_captain", user_player_ids) new_vice_captain = next(p for p in user_picks if p["is_vice_captain"]) assert new_vice_captain["element"] == vice_captain["element"]