Update xls_dslx_fmt_test for std.x after using generics

Author: joannabrozek

xls/dslx/fmt/ast_fmt.cc

@@ -416,6 +416,12 @@ DocRef Fmt(const ChannelTypeAnnotation& n, Comments& comments,
   return ConcatNGroup(arena, pieces);
 }
 
+DocRef Fmt(const TypeVariableTypeAnnotation& n, Comments& comments,
+           DocArena& arena) {
+  std::vector<DocRef> pieces = {Fmt(*n.type_variable(), comments, arena)};
+  return ConcatNGroup(arena, pieces);
+}
+
 DocRef Fmt(const TypeAnnotation& n, Comments& comments, DocArena& arena) {
   if (auto* t = dynamic_cast<const BuiltinTypeAnnotation*>(&n)) {
     return Fmt(*t, comments, arena);
@@ -432,6 +438,12 @@ DocRef Fmt(const TypeAnnotation& n, Comments& comments, DocArena& arena) {
   if (auto* t = dynamic_cast<const ChannelTypeAnnotation*>(&n)) {
     return Fmt(*t, comments, arena);
   }
+  if (auto* t = dynamic_cast<const TypeVariableTypeAnnotation*>(&n)) {
+    return Fmt(*t, comments, arena);
+  }
+  if (dynamic_cast<const GenericTypeAnnotation*>(&n)) {
+    return arena.Make(Keyword::kType);
+  }
   if (dynamic_cast<const SelfTypeAnnotation*>(&n)) {
     return arena.Make(Keyword::kSelfType);
   }

xls/dslx/stdlib/std.x

@@ -19,7 +19,7 @@
 pub fn enumerate<T: type, N: u32>(x: T[N]) -> (u32, T)[N] {
     for (i, result): (u32, (u32, T)[N]) in u32:0..N {
         update(result, i, (i, x[i]))
-    }(((u32, T)[N]: [(u32:0, zero!<T>()), ...]))
+    }(((u32, T)[N]:[(u32:0, zero!<T>()), ...]))
 }
 
 #[test]
@@ -35,7 +35,7 @@ fn emumerate_test() {
 #[test]
 fn enumerate_type_test() {
     type RamData = uN[8];
-    const DATA = RamData[4]: [RamData:1, RamData:2, RamData:4, RamData:8];
+    const DATA = RamData[4]:[RamData:1, RamData:2, RamData:4, RamData:8];
     let enumerated = enumerate(DATA);
     assert_eq(enumerated[u32:0], (u32:0, u8:1));
     assert_eq(enumerated[u32:1], (u32:1, u8:2));