switched diff gen to unit tests