Update test_base.py