Added an actual test to unit_tests.py