"""Tests for training package."""