Update test_codes.py

derp.