Update rizki_test.py

changes