"""Tests for training.yolo module."""