Change these to assertIsNone and modify the category stuff to return None if a canidate isn't in a category.