Add get_unit test.