adding a test using unittest