Modified function test wiith user story and used unittest class approach