move predicted label testing to its own test