Adding tests for unit specification functionality