Implement full set of tests for `Unit` class