Python coding interview

shape