Remove unused functions and add more test coverage

Sorry, being super picky, not clear from the variable name that this is pixel pitch not the image shape