Scala Partial Function Python