reveal_type should work with type_info

Summary:
All we need is to use our existing display, but also make sure that
the `expand` and `for_display` transforms are applied throughout
the `TypeInfo`.

Reviewed By: ndmitchell

Differential Revision: D72496978

fbshipit-source-id: 4160edbc779b74c3c7f9e5c943d56e3d9ccfdaba

Author: stroxler

pyrefly/lib/alt/expr.rs

@@ -596,7 +596,7 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
         self.expr_infer_type_info(x, errors).into_ty()
     }
 
-    fn expr_infer_type_info(&self, x: &Expr, errors: &ErrorCollector) -> TypeInfo {
+    pub fn expr_infer_type_info(&self, x: &Expr, errors: &ErrorCollector) -> TypeInfo {
         self.expr_infer_type_info_with_hint(x, None, errors)
     }
 

pyrefly/lib/alt/special_calls.rs

@@ -24,6 +24,7 @@ use crate::error::kind::ErrorKind;
 use crate::types::callable::unexpected_keyword;
 use crate::types::special_form::SpecialForm;
 use crate::types::types::Type;
+use crate::util::visit::VisitMut;
 
 impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
     pub fn call_assert_type(
@@ -97,13 +98,16 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
         errors: &ErrorCollector,
     ) -> Type {
         if args.len() == 1 {
-            let t = self.solver().deep_force(self.expr_infer(&args[0], errors));
+            let mut type_info = self.expr_infer_type_info(&args[0], errors);
+            type_info.visit_mut(&mut |ty| {
+                *ty = self.for_display(self.solver().deep_force(ty.clone()));
+            });
             self.error(
                 errors,
                 range,
                 ErrorKind::RevealType,
                 None,
-                format!("revealed type: {}", self.for_display(t)),
+                format!("revealed type: {}", type_info),
             );
         } else {
             self.error(