update tests for new format