switched from pytest to unittest