changing the testing to unittest framework