Add tests for the remaining piece types