"""Unit tests for training components"""