Update and rename test_viralize.py to test_controller.py