adding unittest from docs