Functions returning arbitrary types

1 files changed, 5 insertions, 0 deletions

diff --git a/src/Parser.hs b/src/Parser.hs
index 940bd60..1339ec8 100644
--- a/src/Parser.hs
+++ b/src/Parser.hs
@@ -54,6 +54,11 @@ parseDefinition = label "symbol definition" $ toplevel ToplevelDefinition $ do
                         atypes' <- getInferredTypes atypes
                         ( name, ) . SomeExpr . ArgsReq atypes' . FunctionAbstraction <$> replaceDynArgs (mconcat steps)
                 return $ L.IndentSome Nothing finish testStep
+            , do
+                osymbol "="
+                SomeExpr (expr :: Expr e) <- someExpr
+                atypes' <- getInferredTypes atypes
+                L.IndentNone . ( name, ) . SomeExpr . ArgsReq atypes' . FunctionAbstraction <$> replaceDynArgs expr
             ]
     modify $ \s -> s { testVars = fmap someExprType def : testVars s }
     return def