Add tests for test_base