Added unit testing for stock model