writing unit tests for my classes