Test for python class