Using unittest first test

altered the test to use unittest framework.