add unit test to test ImageGenerator