I’m thinking something more along the lines of Haskell or Idris type definitions. You can use that to specify the function signature, and that becomes the contract that the function has to fulfill.