draft of unit tests