Improve docstring of `HasTraits.remove_trait`.
Add tests for `add_trait` and `remove_trait`.