switched from unittest to pytest