unit test for python class