FIX removed stray tests from base.py

Forgot to remove the tests from `base.py`