Add new file test_training.py and test_gui.py