Create test_3.py

Modified task3_final.py for pytest