Normalize union constructors

1 files changed, 12 insertions, 2 deletions

diff --git a/dhall_core/src/printer.rs b/dhall_core/src/printer.rs
index bb3c427..4d1ae2d 100644
--- a/dhall_core/src/printer.rs
+++ b/dhall_core/src/printer.rs
@@ -92,9 +92,9 @@ impl<SE: Display + Clone, N, E: Display> Display for ExprF<SE, Label, N, E> {
                 write!(f, "{} = {}", k, v)
             })?,
             UnionType(a) => fmt_list("< ", " | ", " >", a, f, |(k, v), f| {
-                write!(f, "{} : ", k)?;
+                write!(f, "{}", k)?;
                 if let Some(v) = v {
-                    v.fmt(f)?
+                    write!(f, ": {}", v)?;
                 }
                 Ok(())
             })?,
@@ -108,6 +108,16 @@ impl<SE: Display + Clone, N, E: Display> Display for ExprF<SE, Label, N, E> {
                 }
                 f.write_str(" >")?
             }
+            UnionConstructor(x, map) => {
+                fmt_list("< ", " | ", " >", map, f, |(k, v), f| {
+                    write!(f, "{}", k)?;
+                    if let Some(v) = v {
+                        write!(f, ": {}", v)?;
+                    }
+                    Ok(())
+                })?;
+                write!(f, ".{}", x)?
+            }
             Embed(a) => a.fmt(f)?,
             Note(_, b) => b.fmt(f)?,
         }