ast_cloner: clone test node for conditionals

Signed-off-by: Wojciech Sipak <[email protected]>

Author: wsipak

xls/dslx/frontend/ast_cloner.cc

@@ -1110,8 +1110,10 @@ class AstCloner : public AstNodeVisitor {
       const ConstConditionalTypeAnnotation* n) override {
     XLS_RETURN_IF_ERROR(ReplaceOrVisit(n->consequent_type()));
     XLS_RETURN_IF_ERROR(ReplaceOrVisit(n->alternate_type()));
+    XLS_RETURN_IF_ERROR(ReplaceOrVisit(n->test()));
     old_to_new_[n] = module(n)->Make<ConstConditionalTypeAnnotation>(
-        n->span(), n->test(),
+        n->span(),
+        down_cast<Expr*>(old_to_new_[n->test()]),
         down_cast<TypeAnnotation*>(old_to_new_[n->consequent_type()]),
         down_cast<TypeAnnotation*>(old_to_new_[n->alternate_type()]));
     return absl::OkStatus();