Python Assert example