Updated test_axes.py