Add a preliminarily support for define/contract

sorawee · Jun 5, 2023 · fa8fd5c · fa8fd5c
1 parent 114e1f7
commit fa8fd5c
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 0 deletions.
diff --git a/conventions.rkt b/conventions.rkt
@@ -427,6 +427,8 @@
   [("λ" "lambda") (format-define-like)]
   [("match-define" "match-define-values") (format-define-like)]
 
+  [("define/contract") (format-define-like)]
+
   [("let*") format-let*]
   [("let-values" "let*-values" "letrec" "letrec-values") format-parameterize]
   [("let-syntax" "letrec-syntax" "let-syntaxes" "letrec-syntaxes") format-parameterize]

diff --git a/tests/define-contract.rkt b/tests/define-contract.rkt
@@ -0,0 +1,11 @@
+(define/contract x integer? 1)
+
+(define/contract (f x) (-> integer? integer?)
+  (cond
+    [(zero? x) 1]
+    [else 2]))
+
+(define/contract (f wwwwwwwwwwwwwwww xxxxxxxxxxxxxx yyyyyyyyyyyy zzzzzzzzzzzzzzzzzz) (-> integer? integer? integer?)
+  (cond
+    [(zero? x) 1]
+    [else 2]))
diff --git a/tests/define-contract.rkt.out b/tests/define-contract.rkt.out
@@ -0,0 +1,15 @@
+(define/contract x
+  integer?
+  1)
+
+(define/contract (f x)
+  (-> integer? integer?)
+  (cond
+    [(zero? x) 1]
+    [else 2]))
+
+(define/contract (f wwwwwwwwwwwwwwww xxxxxxxxxxxxxx yyyyyyyyyyyy zzzzzzzzzzzzzzzzzz)
+  (-> integer? integer? integer?)
+  (cond
+    [(zero? x) 1]
+    [else 2]))