Python Assert Called Once