def test_active_inmates_ids(self): inmate_class = Mock() j_ids = [j_id for j_id in range(1, 4)] input_values = [make_county_inmate(j_id) for j_id in j_ids] inmate_class.active_inmates.return_value = input_values inmates = Inmates(inmate_class, Mock()) response_q = Queue(1) inmates.active_inmates_ids(response_q) active_inmates_ids = response_q.get() assert active_inmates_ids == j_ids