Create hw2_test.py