Add unit tests for size