Update test_robustus.py