fix(grainfmt): Use only the correct comments for constraints (#1388)

Use only the correct comments for contraints

Co-authored-by: Marcus Roberts <marcus@marcusr.com>

Author: marcusroberts

compiler/src/formatting/format.re

@@ -2982,7 +2982,15 @@ and print_expression =
           ),
           Doc.text(":"),
           Doc.space,
-          print_type(~original_source, ~comments, parsed_type),
+          print_type(
+            ~original_source,
+            ~comments=
+              Comment_utils.get_comments_inside_location(
+                ~location=parsed_type.ptyp_loc,
+                comments,
+              ),
+            parsed_type,
+          ),
         ]),
       );
     | PExpLambda(patterns, expression) =>

compiler/test/formatter_inputs/constraints.gr

@@ -0,0 +1,8 @@
+import List from "list"
+let test = test => {
+  // Comments
+  List.forEach(i => {
+    // Comment In ForEach
+    print(i)
+  }, test: List<Number>)
+}

compiler/test/formatter_outputs/constraints.gr

@@ -0,0 +1,8 @@
+import List from "list"
+let test = test => {
+  // Comments
+  List.forEach(i => {
+    // Comment In ForEach
+    print(i)
+  }, test: List<Number>)
+}

compiler/test/suites/formatter.re

@@ -45,4 +45,5 @@ describe("formatter", ({test, testSkip}) => {
   assertFormatOutput("parens", "parens");
   assertFormatOutput("windows", "windows");
   assertFormatOutput("patterns", "patterns");
+  assertFormatOutput("constraints", "constraints");
 });