override_feature_check() and override_feature_checks() weren't

properly cleaning up after a crash or assertion, causing a feature flag

to stay to stay modified after use, potentially disrupting other tests.
This change fixes that by ensuring we clear the state in a finally

clause.
It also improves these methods by allowing a Feature instance to be

passed instead of just its ID, which simplifies usage.

Unit tests pass.