1 files changed, 128 insertions, 0 deletions
diff --git a/test/unit/expr/expr_manager_public.h b/test/unit/expr/expr_manager_public.h
new file mode 100644
index 000000000..ecb71081d
--- /dev/null
+++ b/test/unit/expr/expr_manager_public.h
@@ -0,0 +1,128 @@
+/*********************                                                        */
+/** expr_manager_public.h
+ ** Original author: cconway
+ ** Major contributors: none
+ ** Minor contributors (to current version): none
+ ** This file is part of the CVC4 prototype.
+ ** Copyright (c) 2009, 2010  The Analysis of Computer Systems Group (ACSys)
+ ** Courant Institute of Mathematical Sciences
+ ** New York University
+ ** See the file COPYING in the top-level source directory for licensing
+ ** information.
+ **
+ ** Public black-box testing of CVC4::Expr.
+ **/
+
+#include <cxxtest/TestSuite.h>
+
+#include <sstream>
+#include <string>
+
+#include "expr/expr_manager.h"
+#include "expr/expr.h"
+#include "util/Assert.h"
+#include "util/exception.h"
+
+using namespace CVC4;
+using namespace CVC4::kind;
+using namespace std;
+
+class ExprManagerPublic : public CxxTest::TestSuite {
+private:
+
+  ExprManager* d_exprManager;
+
+  void checkAssociative(Expr expr, Kind kind, unsigned int numChildren) {
+    std::vector<Expr> worklist;
+    worklist.push_back(expr);
+
+    unsigned int childrenFound = 0;
+
+    while( !worklist.empty() ) {
+      Expr current = worklist.back();
+      worklist.pop_back();
+      if( current.getKind() == kind ) {
+        for( unsigned int i = 0; i < current.getNumChildren(); ++i ) {
+          worklist.push_back( current.getChild(i) );
+        }
+      } else {
+        childrenFound++;
+      }
+    }
+
+    TS_ASSERT_EQUALS( childrenFound, numChildren );
+  }
+
+  std::vector<Expr> mkVars(Type type, unsigned int n) {
+    std::vector<Expr> vars;
+    for( unsigned int i = 0; i < n; ++i ) {
+      vars.push_back( d_exprManager->mkVar(type) );
+    }
+    return vars;
+  }
+
+
+public:
+  void setUp() {
+    d_exprManager = new ExprManager;
+  }
+
+
+  void tearDown() {
+    try {
+      delete d_exprManager;
+    } catch(Exception e) {
+      cerr << "Exception during tearDown():" << endl << e;
+      throw ;
+    }
+  }
+
+  void testMkAssociative() {
+    try {
+      std::vector<Expr> vars = mkVars(d_exprManager->booleanType(), 294821);
+      Expr n = d_exprManager->mkAssociative(AND,vars);
+      checkAssociative(n,AND,vars.size());
+
+      vars = mkVars(d_exprManager->booleanType(), 2);
+      n = d_exprManager->mkAssociative(OR,vars);
+      checkAssociative(n,OR,2);
+    } catch( Exception& e ) {
+      cerr << "Exception in testMkAssociative(): " << endl << e;
+      throw;
+    }
+  }
+
+  void testMkAssociative2() {
+    try {
+      std::vector<Expr> vars = mkVars(d_exprManager->booleanType(), 2);
+      Expr n = d_exprManager->mkAssociative(OR,vars);
+      checkAssociative(n,OR,2);
+    } catch( Exception& e ) {
+      cerr << "Exception in testMkAssociative2(): " << endl << e;
+      throw;
+    }
+  }
+
+  void testMkAssociative3() {
+    try {
+      unsigned int numVars = d_exprManager->maxArity(AND) + 1;
+      std::vector<Expr> vars = mkVars(d_exprManager->booleanType(), numVars);
+      Expr n = d_exprManager->mkAssociative(AND,vars);
+      checkAssociative(n,AND,numVars);
+    } catch( Exception& e ) {
+      cerr << "Exception in testMkAssociative3(): " << endl << e;
+      throw;
+    }
+  }
+
+  void testMkAssociativeTooFew() {
+    std::vector<Expr> vars = mkVars(d_exprManager->booleanType(), 1);
+    TS_ASSERT_THROWS( d_exprManager->mkAssociative(AND,vars), AssertionException);
+  }
+
+  void testMkAssociativeBadKind() {
+    std::vector<Expr> vars = mkVars(d_exprManager->integerType(), 10);
+    TS_ASSERT_THROWS( d_exprManager->mkAssociative(TUPLE,vars), AssertionException);
+  }
+
+};