Unit test for loading training images