unit tests for two classes