I think that the easiest option to still have a robust test is to check that there is only one UserWarning. Should be simple to code.