Assert Function Called Python Pytest