Update pieces_tests.py