updated tests, probably not in time for iclr