How can I create the picture below using this code? I need to complete the code marked as #TODO
def createTestImg(): img = np.zeros((#TODO, #TODO), dtype