fixed tests.py to use new function names