Refactor bok_choy tests.