Update rizki_test.py