Moved tests to unit_tests.py.