updated test to unittest format
updated setup.py file