Fix thrift tests.

Testing Done:
./pants src/test/python/:all

Reviewed at https://reviews.apache.org/r/21502/