Add new task tests to setup.py