def step_impl(context, count, department): count_ = NamedNumber.from_string(count) assert_that(count_, equal_to(context.model.get_headcount_for(department)))