move ImageModel tests into one class