Adding tests for image size generation