Add screenshot test

Basic test for generating a screenshot.