Update jan_whiff_test.py