Update codebreaker_test.py