Add StratifiedSampler2D test