add tests for unit type