Improve unit-tests.

Поправил unit - тесты
чтобы условие соответсвовало проверке