added tests for wrapper class