Refactoring zkocc_test.py.