Assert Function In Pytest