Update test_iris.py