Add set image tests