Create test_1.py

Modified task1_final.py for pytest