Axis::setUnit(name) now returns new unit

This avoids having to retrieve the new unit immediately after setting
it. Refs #6882
Add a test for unit replacement on an Axis from Python. Refs #6882