add tests to setup.py