use py.test assert