Create main_test.py

Main file for testing