Python function tests now work