Update test_pivot.py