Extending unit tests for AgnInferCDSVisitor class