Adding another unit test to test the deserialized training data