Add masks support to ImageAssertion