ran setup.py successfully
Added test cases