Create assignment_one_test.py