Extend unit test for AgnInferCDSVisitor