Adding back test_base.py, after accidental delete