testing: improve removeImage

Now we support image removal by repository too.