COS moved tests

Moved tests from base.py to tests/test_base.py. Need more tests!