Improve testing

After discussion with Lisa, we figured out some aspects to improve:

• reduce tests and create a new one test_training
• modifications on loading_data according to test_model_load
• fix bugs and restore the functionality in validation_real_data