python wave function collapse