add test file main.py